比较器

比较器

我们知道，使用TreeSet集合当我们存储的元素是有默认的规则时候，取出的元素顺序是升序输出的，那么当我们存储一个自定义的引用类型数据的时候也想让他们按照一个顺序输出怎么办呢？这个时候就要用到我们的比较器了…

假设我们要存入一个 Student 类型的元素到集合：
第一种方法：使用Comparator比较器

class Student{
    private String name;
    private int age;
    private double height;

    @Override
    public String toString() {
        return "Student{" +
                "name='" + name + '\'' +
                ", age=" + age +
                ", height=" + height +
                '}';
    }

    public Student() {
    }

    public Student(String name, int age, double height) {
        this.name = name;
        this.age = age;
        this.height = height;
    }

    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;
    }

    public double getHeight() {
        return height;
    }

    public void setHeight(double height) {
        this.height = height;
    }
}

定义测试类：

public class Test1 {
    public static void main(String[] args) {
        ArrayList<Student> list = new ArrayList<>();
        list.add(new Student("张杰",25,180.0));
        list.add(new Student("鹿晗",27,175.0));
        list.add(new Student("李易峰",22,182.0));
        list.add(new Student("吴亦凡",24,189.0));
         /*
         使用集合的工具类Collections中的Sort(List<T> list, Comparator<? super T> c)方法,
         这个方法有两个参数：一个集合对象，一个比较器对象，现在我们没有比较器对象，那怎么办？
         有两个方法：
            1. 使用匿名内部类
            2. 定义一个类实现我们的Comparator接口，重写compare()方法
         这里我们使用第一种方法：
        */
        Collections.sort(list, new Comparator<Student>() {
            @Override
            public int compare(Student o1, Student o2) {
                return o1.getAge() - o2.getAge();//这里我们让他按照年龄升序来排序
            }
        });
        for (Student student : list) {
            System.out.println(student);
        }
    }
}

第二种方法：让 Student 实现Comparable接口自己去比较

class Student implements Comparable<Student>{ //实现Comparable接口需要重写compareTo()方法
    private String name;
    private int age;
    private double height;

	@Override
    public int compareTo(Student o) {
        return o.getAge() - this.getAge();//这里我们降序输出吧
    }

    @Override
    public String toString() {
        return "Student{" +
                "name='" + name + '\'' +
                ", age=" + age +
                ", height=" + height +
                '}';
    }

    public Student() {
    }

    public Student(String name, int age, double height) {
        this.name = name;
        this.age = age;
        this.height = height;
    }

    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;
    }

    public double getHeight() {
        return height;
    }

    public void setHeight(double height) {
        this.height = height;
    }

}

定义测试类：

public class Test2 {
    public static void main(String[] args) {
        ArrayList<Student> list = new ArrayList<>();
        list.add(new Student("张杰",25,180.0));
        list.add(new Student("鹿晗",27,175.0));
        list.add(new Student("李易峰",22,182.0));
        list.add(new Student("吴亦凡",24,189.0));

        Collections.sort(list);//这个重载的方法只有一个参数
        for (Student student : list) {
            System.out.println(student);
        }
    }
}