文章目录
前言
TreeSet集合的特点可以将元素按照规则进行排序,这篇笔记里的两种排序方式通常是实现TreeSet集合排序的常用规则。
提示:以下是本篇文章正文内容,下面案例可供参考
一、自然排序
自然排序,就是让元素所属的类实现Comparable接口,重写compareTo(T o)方法
1.自然排序Comparable的实现
用TreeSet集合存储自定义对象,无参构造方法使用的是自然排序对元素进行排序的。自定义的类就必须要实现Comparable接口,重写compareTo(T o)方法,重写compareTo方法时一定要注意排序规则必须按照要求的主要条件和次要条件来写。
2.案例实现
-
案例需求
-
存储学生对象并遍历,创建TreeSet集合使用无参构造方法
-
要求:按照年龄从小到大排序
代码如下:
学生类
public class Student implements Comparable<Student>{
private String name;
private int age;
public Student() {
}
public Student(String name, int age) {
this.name = name;
this.age = age;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public int getAge() {
return age;
}
public void setAge(int age) {
this.age = age;
}
@Override
public String toString() {
return "Student{" +
"name='" + name + '\'' +
", age=" + age +
'}';
}
/***
* 比较两个对象的大小
* @param o 表示集合中的元素
* @return 如果是正值,表示要存入的当前对象大,放右边
* 如果是负值,表示要存入的当前对象小,入左边
* 如果返回0, 表示两个对象相等,不存
*/
@Override
public int compareTo(Student o) {
//按照对象的年龄进行排序
//主要判断条件
int result = this.age - o.age;//年龄小的放前面,年龄大的放后面,从小到大排
return result;
}
}
测试类
代码如下:
public class MyTreeSet2 {
public static void main(String[] args) {
//创建集合对象
TreeSet<Student> ts = new TreeSet<>();
//创建学生对象
Student s1 = new Student("zhangsan",28);
Student s2 = new Student("lisi",27);
Student s3 = new Student("wangwu",29);
Student s4 = new Student("zhaoliu",28);
Student s5 = new Student("qianqi",30);
//把学生添加到集合
ts.add(s1);
ts.add(s2);
ts.add(s3);
ts.add(s4);
ts.add(s5);
//遍历集合
for (Student student : ts) {
System.out.println(student);
}
}
}
二、比较器排序Comparator
比较器排序,就是让集合构造方法接收Comparator的实现类对象,重写compare(T o1,T o2)方法。
int compare(T o1,T o2)
比较其两个参数的顺序。 返回负整数,零或正整数,因为第一个参数小于,等于或大于第二个参数
1.比较器排序Comparator的实现
-
用TreeSet集合存储自定义对象,带参构造方法使用的是比较器排序对元素进行排序的
-
比较器排序,就是让集合构造方法接收Comparator的实现类对象,重写compare(T o1,T o2)方法
-
重写方法时,一定要注意排序规则必须按照要求的主要条件和次要条件来写
2.案例实现
案例需求
- 存储老师对象并遍历,创建TreeSet集合使用带参构造方法
- 要求:按照年龄从小到大排序,年龄相同时,按照姓名的字母顺序排序
代码如下(示例):
老师类
public class Teacher {
private String name;
private int age;
public Teacher() {
}
public Teacher(String name, int age) {
this.name = name;
this.age = age;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public int getAge() {
return age;
}
public void setAge(int age) {
this.age = age;
}
@Override
public String toString() {
return "Teacher{" +
"name='" + name + '\'' +
", age=" + age +
'}';
}
}
public class MyTreeSet4 {
public static void main(String[] args) {
TreeSet<Teacher> ts = new TreeSet<>(
new Comparator<Teacher>() {
/***
* 比较两个对象的大小
* @param o1 表示要存入的元素
* @param o2 表示已经存入到集合中的元素
* @return 如果是正值,表示要存入的当前对象大,放右边
* 如果是负值,表示要存入的当前对象小,入左边
* 如果返回0, 表示两个对象相等,不存
*/
@Override
public int compare(Teacher o1, Teacher o2) {
//o1表示现在要存入的那个元素
//o2表示已经存入到集合中的元素
//主要条件
int result = o1.getAge() - o2.getAge();
//次要条件
result = result == 0 ? o1.getName().compareTo(o2.getName()) : result;
return result;
}
});
Teacher t1 = new Teacher("zhangsan",23);
Teacher t2 = new Teacher("lisi",22);
Teacher t3 = new Teacher("wangwu",24);
Teacher t4 = new Teacher("zhaoliu",24);
ts.add(t1);
ts.add(t2);
ts.add(t3);
ts.add(t4);
System.out.println(ts);
}
}
三、两种排序方式的对比
1.不同点
1.用到的接口不同
自然排序: 自定义类实现Comparable接口,重写compareTo方法,根据返回值进行排序
比较器排序: 创建TreeSet对象的时候传递Comparator的实现类对象,重写compare方法,根据返回值进行排序
2.使用场景不同
自然排序能满足大部分情况,并且java中默认提供的类如String、Integer都默认使用的是自然排序。
当自然排序不满足需要时(也就是说没法重写compareTo方法时),例如:改变包装类Integer的比较规则,让其从大到小,没办法去改源码的compareTo的比较规则,就可以再在创建集合对象时传入一个比较器对象并自己重写compare制定排序规则。
2.相同点
返回值的规则:
- 如果返回值为负数,表示当前存入的元素是较小值,存左边
- 如果返回值为0,表示当前存入的元素跟集合中元素重复了,不存
- 如果返回值为正数,表示当前存入的元素是较大值,存右边