Comparable接口和 Comparator接口的使用

系统管理员 2022-05-28 12:55 372阅读 0赞

Comparable接口 和 Comparator接口 都是可以对集合对象进行排序的， 使用方式分别如下。

## Comparable ##

假如现在有一个学生类， 要以学生年龄从小到大排序， 这时候可以在Student类上实现Comparable接口, compareTo方法比较两个对象的年龄， 1表示大于，-1表示小于，0为相等。

@Data
    @AllArgsConstructor
    @NoArgsConstructor
    public class Student implements Comparable<Student> { 
        private String name;
        private Integer age;
    
        @Override
        public int compareTo(Student o) {
            if(this.age > o.age){
                return 1;
            }
            if (this.age < o.age){
                return -1;
            }
            return 0;
        }
    }

创建两个学生对象, 并比较, 经过compareTo方法比较得到-1.

Student s1 = new Student("a", 12);
        Student s2 = new Student("b",19);
        System.out.println(s1.compareTo(s2));

接着准备数据集合

List<Student> list = new ArrayList<Student>();
        list.add(new Student("ccc", 12));
        list.add(new Student("aaa",17));
        list.add(new Student("bbb",11));
        list.add(new Student("ddd",19));

现在的list中是没有顺序的，我们可以用 Collections.sort() 完成对list的排序。

Collections.sort(list);
    list.forEach((Student stu) -> System.out.println(stu));

## Comparator ##

当Student类没有实现Comparable接口我们就可以用Comparator接口来比较两个Student对象。

Student类， 不实现Comparable

@Data
    @AllArgsConstructor
    @NoArgsConstructor
    public class Student{ 
        private String name;
        private Integer age;
    }

对list排序, 其中compare规则和compareTo一样

Collections.sort(list, new Comparator<Student>() {
            @Override
            public int compare(Student o1, Student o2) {
                if(o1.getAge() > o2.getAge()){
                    return 1;
                }
                if (o1.getAge() < o2.getAge()){
                    return -1;
                }
                return 0;
            }
        });
    
        list.forEach((Student stu) -> System.out.println(stu));

使用lambda可以消除内部类的多余代码

Collections.sort(list, (s1, s2) -> s1.getAge() > s2.getAge() ? 1: -1);