前言
当前有个需求,有一个User实体类,我想按照年龄对User类进行排序,年龄一样需要按照身高再进行排序,该怎么做?答案:可以让User类实现Comparable接口,排序规则定义在重写的方法里面。
引导
- 我们平时对List进行排序,只需调用Collections.sort(list)方法,就能实现对list的排序。这貌似天经地义,他就能实现排序。
@Test
public void myTest() {
ArrayList<Integer> list = new ArrayList<>();
list.add(2);
list.add(5);
list.add(4);
list.add(1);
list.add(3);
Collections.sort(list);
for (Integer integer : list) {
System.out.println(integer);
}
}
实现对User类的age进行排序
- 那么我们要对User对象里面的年龄进行排序呢,我们直接创建个student对象,看看能不能实现排序
User类
@Data
@AllArgsConstructor
public class User {
String name;
int age;
}
测试方法
报错了!我们来看看sort方法里面怎么定义的
哦,明白了,我们定义的ArrayList<User> list,list的泛型是User,我们的User没有继承Comparable,所以不能排序。要想排序,我们只需改动User类,让他继承Comparable。
有人会问,刚开始list插入的数字,测试不是成功了吗,怎么加入自定义的实体类就不行了呢?我们来看下刚才的代码,我们定义的是ArrayList<Integer> list,list的泛型是Integer。通过源码发现,Integer已经实现了comparable方法。除此之外,像Long、String、Double这些基础类类型的包装类型,都已经早早的实现了comparable方法,可以直接对他们排序。
最终实现comparable的User类
- 我习惯上用当前对象的值 减去 传入对象的值: 即 this.age - user.getAge()
- 很多人会问,这里返回1 -1 0 都是啥意思?其实你重写的compareTo方法,现在不会执行,只有在排序的时候才会被调用,排序算法不就那么几种吗,你可以理解为冒泡(方便理解,真正并非冒泡排序)。你比较前后两个值大小是不是需要两个数相减,根据大小决定要交换还是原地不动。你这里写的comparabTo方法,就在那个时候被多次调用。太抽象?我给您举个栗子吧,请看:
@Data
@AllArgsConstructor
public class User implements Comparable<User>{
String name;
int age;
@Override
public int compareTo(User user) {
//从小到大等价写法1:
if (this.age > user.getAge()){
return 1;
}else if (this.age < user.getAge()){
return -1;
}else {
return 0;
}
//从小到大等价写法二:
// return this.age - user.getAge();
}
}
测试类
测试结果
需求变了,现在我们要求按年龄从小到大排,年龄一样的时候,按照身高从小到大排。
- 此时只需改动comparaTo方法
@Data
@AllArgsConstructor
public class User implements Comparable<User> {
String name;
int age;
int lengh;
@Override
public int compareTo(User user) {
//从小到大等价写法1:
if (this.age > user.getAge()) {
return 1;
} else if (this.age < user.getAge()) {
return -1;
} else {//此时就是年龄都一样的情况下会进来
if (this.lengh > user.getLengh()){
return 1;
}else if(this.lengh < user.getLengh()){
return -1;
}else {
return 0;
}
}
//从小到大等价写法二:
// if (this.age > user.getAge()) {
// return 1;
// } else if (this.age < user.getAge()) {
// return -1;
// } else {//此时就是年龄都一样的情况下会进来
// return this.age - user.getAge();
// }
}
}
测试
- 完全正确