自然排序Comparable的使用

Posted 2021-03-02 pxy-1999

首先说明用TreeSet集合存储自定义时，无参构造方法使用的是自然排序对元素进行排序的

 

然后，使用自然排序的前提是，让元素所属的类实现Comparable接口并重写comparaTo(T o)方法，来完成。

 

其次要注意，重写方法时，一定要注意排序规则必须按照要求的主要条件和条件来写。 下面通过一个需求来具体实现一下。

需求：存储学生对象并遍历，创建TreeSet集合使用无参构造方法。

要求：按照年龄从小到大排序，年龄相同时，按照姓名的字母顺序排序

分析可知，需要两个类：一个学生类（Student）一个操作类（TreeSetDemo）学生类中有两个成员遍历：姓名（name）、年龄（age）。（仅记录重写代码部分）

操作类代码：

public class TreeSetDemo {
    public static void main(String[] args) {
        //创建集合对象
        TreeSet<Student> treeSet = new TreeSet<Student>();

        //创建学生对象
        Student s1 = new Student("xishi",29);
        Student s2 = new Student("wangzhaojun",30);
        Student s3 = new Student("diaocan",28);
        Student s4 = new Student("yangyuhuan",31);

        //添加元素
        treeSet.add(s1);
        treeSet.add(s2);
        treeSet.add(s3);
        treeSet.add(s4);

        //遍历
        for (Student s : treeSet){
            System.out.println(s);
        }
    }
}

接下来测试，在学生类中实现了Comparable<E>接口，并重写compareTo(T o)方法后，操作类的运行情况

在Student类中重写compareTo(T o)方法代码：（通过返回值的不同，来影响排序结果，具体看运行结果）

@Override
    public int compareTo(Student o) {
        return 0;
    }
    /*
        运行结果：xishi,29
     */

@Override
    public int compareTo(Student o) {
        return 1;
    }
    /*
        运行结果：
                xishi,29
                wangzhaojun,30
                diaocan,28
                yangyuhuan,31
     */

@Override
    public int compareTo(Student o) {
        return -1;
    }
    /*
        运行结果：
                yangyuhuan,31
                diaocan,28
                wangzhaojun,30
                xishi,29
     */

通过三次修改返回值，从运行结果中可以得出结论：

当返回值为0时，认为是重复元素，不进行存储。

当返回值为1时，按照存储顺序，正向排序（升序）（因为是正向排序，可以理解从小到大的顺序）

当返回值为-1时，按照存储顺序，逆向排序（降序）（同理）

于是，可以通过控制返回值的不同来控制不同的排序结果。接下来实现上面的需求。

对compareTo方法在进行修改，先使得年龄以升序的方式排序，代码：

 

@Override
    public int compareTo(Student s) {
        int num = this.age - s.age;
        return num;
    }
    /*
        运行结果：
                diaocan,28
                xishi,29
                wangzhaojun,30
                yangyuhuan,31
     */

 

接下来完善要求，年龄相同时，按照姓名的字母顺序排序，完整代码如下：

//操作类代码
public class TreeSetDemo {
    public static void main(String[] args) {
        //创建集合对象
        TreeSet<Student> treeSet = new TreeSet<Student>();

        //创建学生对象
        Student s1 = new Student("xishi",29);
        Student s2 = new Student("wangzhaojun",30);
        Student s3 = new Student("diaocan",28);
        Student s4 = new Student("yangyuhuan",31);
        Student s5 = new Student("linqingxia",31);

        //添加元素
        treeSet.add(s1);
        treeSet.add(s2);
        treeSet.add(s3);
        treeSet.add(s4);
        treeSet.add(s5);

        //遍历
        for (Student s : treeSet){
            System.out.println(s.getName()+","+s.getAge());
        }
    }
}

//学生类中重写代码部分及运行结果
@Override
    public int compareTo(Student s) {
        int num = this.age - s.age;
        //如果年龄相同则比较姓名字母顺序(升序)排列。
        int num2 = num == 0 ? this.name.compareTo(s.name) : num;
        return num2;
    }
    /*
        运行结果：
                diaocan,28
                xishi,29
                wangzhaojun,30
                linqingxia,31
                yangyuhuan,31
     */