Java如何設計自定義排序器

在使用Java無序集合時，經常需要對集合進行排序，此時需要我們自己去實現排序邏輯。接下來就以TreeSet為例來看看如何對集合進行排序。

TreeSet對元素排序有兩種方式：

• 第一種：復寫Comparable接口的compareTo方法。
• 第二種：采用自定義Comparator比較器

場景：對象Person包含姓名name和年齡age兩個屬性，按照年齡進行升序排序，如果年齡一致，則按照姓名升序排序。

一、復寫Comparable接口的compareTo方法

讓元素自身具備比較功能，元素就需要實現comparable接口，覆蓋compareTo方法。

代碼如下：

class Person implements Comparable{

    private String name;
    private int age;

    // 重寫toString()方法，輸出對象時輸出格式為：name:age
    @Override
    public String toString() {
        return name+ ":" + 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;
    }

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

    @Override
    public int compareTo(Object o) {
        Person p = (Person) o;

        // 根據年齡升序排序
      int temp = this.age - p.age;
      return temp == 0 ? this.name.compareTo(p.name):temp;
    }
}

測試：

public class TreeSetDemo {
    public static void main(String[] args) {
        TreeSet<Person> ts = new TreeSet();
        ts.add(new Person("zhangsan", 20));
        ts.add(new Person("lisi", 20));
        ts.add(new Person("wangwu", 21));
        ts.add(new Person("zhouqi", 29));
        ts.add(new Person("zhaoliu", 28));

        for (Person person : ts) {
            System.out.println(person);
        }
    }
}

結果輸出：

lisi:20
zhangsan:20
wangwu:21
zhaoliu:28
zhouqi:29

但是如果不要按照對象中具備的自然順序進行排序，即對象中不具備自然順序該如何處理呢？就需要采用下面的第二種方式。

 二、采用自定義Comparator比較器

讓集合自身具備比較功能，即在集合對象創建時，由此想到TreeSet的構造方法: new TreeSet(Comparator<? super E> comparator)。為此需要定義一個類實現Comparator接口，覆蓋compare方法，將該類對象作為參數傳遞給TreeSet集合的構造函數。

代碼如下：

PersonNoComparator類不用復寫Comparable接口的compareTo方法。

class PersonNoComparator {

    private String name;
    private int age;

    @Override
    public String toString() {
        return name+":" + 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;
    }

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

設計自定義排序器，並復寫compare方法。

class ComparatorByAge implements Comparator {

    // 根據年齡升序排序
    @Override
    public int compare(Object o1, Object o2) {
        Person p1 = (Person) o1;
        Person p2 = (Person) o2;

        int tmp = p1.getAge() - p2.getAge();
        return tmp == 0 ? p1.getName().compareTo(p2.getName()) : tmp;
    }
}

測試：

與第一種方式測試不同的地方是，在構造TreeSet方法時直接按照ComparatorByAge排序器排序。

public class TreeSetDemo {
    public static void main(String[] args) {
        ts = new TreeSet(new ComparatorByAge());
        ts.add(new Person("zhangsan", 20));
        ts.add(new Person("lisi", 20));
        ts.add(new Person("wangwu", 21));
        ts.add(new Person("zhouqi", 29));
        ts.add(new Person("zhaoliu", 28));

        for (Person person : ts) {
            System.out.println(person);
        }
    }
}

結果輸出：

lisi:20
zhangsan:20
wangwu:21
zhaoliu:28
zhouqi:29

可以發現，兩種方式實現的排序結果是一致的。

 

【參考資料】

1. https://www.jianshu.com/p/9334567d0912