TreeSet集合的自然排序與比較器排序、Comparable接口的compareTo()方法

【自然排序】

package com.hxl;

public class Student implements Comparable<Student> {

    private String name;
    private int age;

    public Student() {
        super();
    }

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

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

    @Override
    public int compareTo(Student s) {
        // 先讓兩個對象的age屬性做差比較，這個是主要排序條件
        int num = this.age - s.age;
        // 若age屬性相同，再比較name屬性(String類本身實現了Comparable接口)
        // 即在主要排序條件相同的情況下，次要排序條件起作用
        int flag = num == 0 ? this.name.compareTo(s.name) : num;
        // 返回比較結果
        return flag;
    }
}

package com.hxl;

import java.util.TreeSet;

public class Test {
    public static void main(String[] args) {
        //這里使用的無參構造實例化TreeSet集合，則默認啟用的是自然排序
        TreeSet<Student> ts = new TreeSet<Student>();
        ts.add(new Student("cc", 11));
        ts.add(new Student("ee", 11));
        ts.add(new Student("cc", 22));
        ts.add(new Student("aa", 22));
        ts.add(new Student("bb", 11));

        for (Student s : ts) {
            System.out.println(s.getName()+"_"+s.getAge());
        }

        /*
            為什么TreeSet集合中的元素既唯一又有序呢？
            原因是它在存儲元素的時候就是有序存儲的（紅黑樹結構存儲）
            TreeSet的add()方法底層依賴的是Comparable的compareTo方法
            這里就是說元素類本身要有自己的compareTo方法
            所以元素類本身必須實現Comparable接口，重寫compareTo方法
            compareTo方法有個特點：它返回的是int型數據，結果有三類負數、0、正數
            例如：(Java中一些常見的有比較意義的一些類都實現了Comparable接口，如Integer類)
                Integer a = new Integer(10);
                Integer b = new Integer(20);
                int num = a.compareTo(b);     //因為a小於b，所以num返回的是負數
            而TreeSet的add()方法這樣理解此返回值：
            即返回負數則比根節點小，元素在此集合中唯一，元素存放根的左孩子
            返回正數則比根節點大，元素在此集合中唯一，元素存放根的右孩子
            返回0則表示，元素相同，在此集合中不唯一，故而丟掉不存放
            由此可見，我們的重寫的compareTo()方法決定了TreeSet集合中元素的去留和順序！
        */
    }
}

【比較器排序（外部類實現）】

package com.hxl;

public class Student{

    private String name;
    private int age;

    public Student() {
        super();
    }

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

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

package com.hxl;

import java.util.Comparator;

public class MyComparator implements Comparator<Student> {

    @Override
    public int compare(Student s1, Student s2) {
        // 先讓兩個對象的age屬性做差比較，這個是主要排序條件
        int num = s1.getAge() - s2.getAge();
        // 若age屬性相同，再比較name屬性(String類本身實現了Comparable接口)
        // 即在主要排序條件相同的情況下，次要排序條件起作用
        int flag = num == 0 ? s1.getName().compareTo(s2.getName()) : num;
        // 返回比較結果
        return flag;
    }
}

package com.hxl;

import java.util.TreeSet;

public class Test {
    public static void main(String[] args) {
        //這里使用TreeSet(Comparator comparator)構造實例化TreeSet集合，則啟用的是指定比較器排序
        TreeSet<Student> ts = new TreeSet<Student>(new MyComparator());
        ts.add(new Student("cc", 11));
        ts.add(new Student("ee", 11));
        ts.add(new Student("cc", 22));
        ts.add(new Student("aa", 22));
        ts.add(new Student("bb", 11));

        for (Student s : ts) {
            System.out.println(s.getName()+"_"+s.getAge());
        }
    }
}

【比較器排序（內部類實現，如果只使用一次的話）】

package com.hxl;

public class Student{

    private String name;
    private int age;

    public Student() {
        super();
    }

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

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

package com.hxl;

import java.util.Comparator;
import java.util.TreeSet;

public class Test {
    public static void main(String[] args) {
        //如果一個方法的參數是接口，那么真實想要的是其實是接口實現類的對象
        //這里的對象只用一次，專門定義一個外部類顯得麻煩
        //匿名內部類可以實現這個需求
        TreeSet<Student> ts = new TreeSet<Student>(new Comparator<Student>(){
            public int compare(Student s1, Student s2) {
                // 先讓兩個對象的age屬性做差比較，這個是主要排序條件
                int num = s1.getAge() - s2.getAge();
                // 若age屬性相同，再比較name屬性(String類本身實現了Comparable接口)
                // 即在主要排序條件相同的情況下，次要排序條件起作用
                int flag = num == 0 ? s1.getName().compareTo(s2.getName()) : num;
                // 返回比較結果
                return flag;
            }
        });
        ts.add(new Student("cc", 11));
        ts.add(new Student("ee", 11));
        ts.add(new Student("cc", 22));
        ts.add(new Student("aa", 22));
        ts.add(new Student("bb", 11));

        for (Student s : ts) {
            System.out.println(s.getName()+"_"+s.getAge());
        }
    }
}

 【注】開發中會用最后一種，因為第一種只有固定的排序方式，第二種每次都要定義外面類顯得麻煩。