Demo:
package jihe;
import java.util.Comparator;
import java.util.TreeSet;
/**
* set 集合中的 TreeSet集合
*
* 需求：
* 使用该集合存储多个学生的考试成绩信息（姓名、学号、语文成绩、数学成绩、历史成绩）
*
* 要求：
* 按照总分降序排序
*/
public class TreeSetDemo {
public static void main(String[] args) {
Student s1 = new Student("张三", 95, 88, 76);
Student s2 = new Student("李四", 67, 93, 84);
Student s3 = new Student("王五", 88, 99, 77);
Student s4 = new Student("赵六", 63, 75, 99);
Student s5 = new Student("田七", 71, 66, 98);
//自然排序
nature(s1, s2, s3, s4, s5);
System.out.println("\n\r");
//比较器排序
comp(s1,s2,s3,s4,s5);
}
/**
* 自然排序
*
* @param s1
* @param s2
* @param s3
* @param s4
* @param s5
*/
public static void nature(Student s1, Student s2, Student s3, Student s4, Student s5) {
TreeSet<Student> st = new TreeSet<>();
st.add(s1);
st.add(s2);
st.add(s3);
st.add(s4);
st.add(s5);
for (Student s : st) {
System.out.println("姓名:" + s.getName() + ",语文分数：" + s.getLangen() + ",数学分数：" + s.getMath() + ",历史分数：" + s.getHistory() + ",总分成绩：" + s.getTotal());
}
}
/**
* 比较强排序
*
* @param s1
* @param s2
* @param s3
* @param s4
* @param s5
*/
public static void comp(Student s1, Student s2, Student s3, Student s4, Student s5) {
//匿名对象
TreeSet<Student> st = new TreeSet<>(new Comparator<Student>() {
/**
* 重写compareTo方法 实现自定义排序
* @param o1 上一个元素
* @param o2 当前元素
* @return int 0、正数、负数
* 返回 0 表示元素已存在，不会存储
* 返回 正数 表示元素插入后面
* 返回 负数 表示元素插入前面
*/
@Override
public int compare(Student o1, Student o2) {
//这里降序排序
int i = o1.getTotal() > o2.getTotal() ? -1 : 1;
return i;
}
});
st.add(s1);
st.add(s2);
st.add(s3);
st.add(s4);
st.add(s5);
for (Student s : st) {
System.out.println("姓名:" + s.getName()+",语文分数："+s.getLangen() + ",数学分数："+s.getMath() + ",历史分数："+s.getHistory()+",总分成绩："+s.getTotal());
}
}
}
Student类:
package jihe;
/**
* 实现排序功能需要实现 Comparable 接口，并且指定 泛型 为需要操作的 引用数据 类型
*/
public class Student implements Comparable<Student> {
String name;//姓名
int langen;//语文成绩
int math;//数学成绩
int history;//历史成绩
public Student(String name, int langen, int math, int history) {
this.name = name;
this.langen = langen;
this.math = math;
this.history = history;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public int getLangen() {
return langen;
}
public void setLangen(int langen) {
this.langen = langen;
}
public int getMath() {
return math;
}
public void setMath(int math) {
this.math = math;
}
public int getHistory() {
return history;
}
public void setHistory(int history) {
this.history = history;
}
/**
* 获取学生总分
*
* @return
*/
public int getTotal() {
return langen + math + history;
}
/**
* 重写compareTo方法 实现自定义排序
*
* @param s 需要比较 的对象
* @return int 0、正数、负数
* 返回 0 表示元素已存在，不会存储
* 返回 正数 表示元素插入后面
* 返回 负数 表示元素插入前面
*/
@Override
public int compareTo(Student s) {
//这里升序排序
int i = this.getTotal() > s.getTotal() ? 1 : -1;
return i;
}
}
NOTE:
- 由于Set集合的不保证排序方式，因此在Set集合的子类TreeSet类中提供了两种排序方式
- 自然排序和比较强排序
- 自然排序是在操作的对象中实现 Comparable 接口 （implements Comparable
） - 比较器排序是在操作的对象实例化时提供构造参数 Comparable （implements Comparable
）
