TreeSet 的两种排序方式 Comparable Comparator

 Comparator 

Student 类

package com.gao;

public class Student implements Comparable<Student>{
    private int age;
    private String name;

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

    public int getAge() {
        return age;
    }

    public void setAge(int age) {
        this.age = age;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

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

    @Override
    public int compareTo(Student o) {
        //我想要按照 年龄的升序进行排序
        int result = this.getAge() - o.getAge();
        return result;
    }
}

测试方法：

package com.gao;

import java.util.TreeSet;

public class xiangshang {
    public static void main(String[] args) {
        Student san = new Student(15, "san");
        Student si = new Student(18, "san");
        Student wu = new Student(5, "san");

        TreeSet<Student> students = new TreeSet<>();
        students.add(san);
        students.add(si);
        students.add(wu);

        for (Student student : students) {
            System.out.println(student);
        }


    }
}

结果：按照年龄排序

法一：实现Comparable 方法：重写compareTo方法。在插入元素的时候不断的调用compareTo方法，所以在插入的时候就确定了顺序。

黑马的解释：

    @Override
    public int compareTo(Student o) {
        //我想要按照 年龄的升序进行排序
        int result = this.getAge() - o.getAge();
        return result;
    }

 

换一下顺序就变成了：降序

结果：

 总结：this在前面是升序排序

 法二：比较器排序 

需求：

代码：

TreeSet<String> ts = new TreeSet<>();

        ts.add("c");
        ts.add("ab");
        ts.add("df");
        ts.add("qwer");

        for (String h : ts) {
            System.out.println(h);
        }

 默认排序结果：是按照首字母进行排序的  ，这不是我们想要的结果。

 刚刚装的是student，是实现了Comparable ，我们设计的是按照age  进行排序，但是现在，我们存储的String，String类型也是实现了Comparable方法的。

重写了方法：

 这样是按照首字母进行排序的。这不是我们想要的排序方式。

因此我们想到了定制排序。

看下  TreeSet 的构造方法吧。

/**
             * o1 表示的是当前要添加的元素
             * o2 表示的是红黑树存在的元素
             *
             * 返回值的规则和之前的是一样的
             *
             * new  的是一个匿名内部类,匿名内不内，即使一个接口 也是一个对象，实现的是 Comparator
             * 接口 ，实现接口就需要重写方法，所以重写了compare方法。
             */
            TreeSet<String> ts = new TreeSet<>(new Comparator<String>() {
                @Override
                public int compare(String o1, String o2) {
                    int result = o1.length() - o2.length();
                    if (result == 0){
                        return o1.compareTo(o2);//如果长度一样，就按照首字母的规则排序
                    }
                    return result;
                }
            });

        ts.add("c");
        ts.add("ab");
        ts.add("df");
        ts.add("qwer");

        for (String h : ts) {
            System.out.println(h);
        }

或者可以简化一下

总结：