指定TreeSet排序规则的两种方法
- 第一种:让元素的类实现Comparable接口,重写compareTo方法
- 第二种:在创建TreeSet集合时,通过构造方法传递Compartor比较器对象
方法一
/第一步:先让Student类,实现Comparable接口
//注意:Student类的对象是作为TreeSet集合的元素的
public class Student implements Comparable<Student>{
private String name;
private int age;
private double height;
//无参数构造方法
public Student(){}
//全参数构造方法
public Student(String name, int age, double height){
this.name=name;
this.age=age;
this.height=height;
}
//...get、set、toString()方法自己补上..
//第二步:重写compareTo方法
//按照年龄进行比较,只需要在方法中让this.age和o.age相减就可以。
/*
原理:
在往TreeSet集合中添加元素时,add方法底层会调用compareTo方法,根据该方法的
结果是正数、负数、还是零,决定元素
放在后面、前面还是不存。
*/
@Override
public int compareTo(Student o) {
//this:表示将要添加进去的Student对象
//o: 表示集合中已有的Student对象
return this.age-o.age;
}
}
方法二:
public static void main(String[] args) {
//创建TreeSet集合时,传递比较器对象排序
/*
原理:当调用add方法时,底层会先用比较器,根据Comparator的compare方是正数、负数、还是零,决定谁在后,谁在前,谁不存。
*/
Set<Student> students = new TreeSet<>(new Comparator<Student>() {
@Override
public int compare(Student o1, Student o2) {
// 下面两句相同,但是调用函数能排除更多的复杂情况,能进行验证类似的
//return (int)(o1.getHeight() - o2.getHeight());
return Double.compare(o1.getAge(),o2.getAge());
}
});
Student s1 = new Student("张三", 18, 180);
Student s2 = new Student("张三", 18, 180);
Student s3 = new Student("张三2", 35, 178);
Student s4 = new Student("李四", 20, 160);
students.add(s1);
students.add(s2);
students.add(s3);
students.add(s4);
System.out.println(students);
}
总结
如果两种比较方法都有,则 第二种 :在创建TreeSet集合时,通过构造方法传递Compartor比较器对象 方法优先级更高