Java集合排序功能實現分析

Java如何實現集合的排序？

- 本文以對Student對象集合為例進行排序
Java通過Collections.sort(List<Student> stuList)和Collections.sort(List<Student> stuList,Comparator c)兩種方法實現排序。

用Collections.sort(List list) 方法實現排序：

step1: 確保Student類實現了Comparable接口，並重寫了compareTo()方法。

step2:調用Collections.sort(List list) 方法進行排序。

public class Student implements Comparable<Student> {

    private int age;

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

    public int getAge() {
        return age;
    }

    @Override
    public int compareTo(Student student) {  // 重寫compareTo方法

        return (this.age < student.age) ? -1 : ((this.age == student.age) ? 0 : 1);
    }


    public static void main(String[] args) {
        List<Student> stuList = new ArrayList();
        stuList.add(new Student(5));
        stuList.add(new Student(3));
        stuList.add(new Student(7));
        stuList.add(new Student(2));
        stuList.add(new Student(4));
        stuList.add(new Student(6));
        stuList.add(new Student(1));

        Collections.sort(stuList);  // 調用排序方法

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

原理分析：

step1: Collections類調用List.sort(Comparator c)方法，比較器c賦值為null.

    public static <T extends Comparable<? super T>> void sort(List<T> list) {
        list.sort(null);
    }

step2: List接口中的sort方法將stuList集合轉換成數組，通過Arrays.sort()方法對其進行排序，並將排序后的元素替換stuList中每個元素。

   default void sort(Comparator<? super E> c) {
        Object[] a = this.toArray();
        Arrays.sort(a, (Comparator) c);
        ListIterator<E> i = this.listIterator();
        for (Object e : a) {
            i.next();
            i.set((E) e);
        }
    }

那到底時在哪里調用的compareTo方法的呢？

進入Arrays.sort()方法：

    public static <T> void sort(T[] a, Comparator<? super T> c) {
        if (c == null) {
            sort(a);
        } else {
            if (LegacyMergeSort.userRequested)
                legacyMergeSort(a, c);
            else
                TimSort.sort(a, 0, a.length, c, null, 0, 0);
        }
    }

沒有制定比較器，因此c==null為true，執行sort(a)方法：

public static void sort(Object[] a) {
        if (LegacyMergeSort.userRequested)
            legacyMergeSort(a);
        else
            ComparableTimSort.sort(a, 0, a.length, null, 0, 0);
    }

LegacyMergeSort.userRequested默認為false，表示是否使用傳統歸並排序，傳統歸並排序在1.5及之前是默認排序方法，1.5之后默認執行ComparableTimSort.sort()方法。除非程序中強制要求使用傳統歸並排序。語句如下：

System.setProperty("java.util.Arrays.useLegacyMergeSort", "true"); 

所以繼續看ComparableTimSort.sort()方法：

    static void sort(Object[] a, int lo, int hi, Object[] work, int workBase, int workLen) {
        assert a != null && lo >= 0 && lo <= hi && hi <= a.length;

        int nRemaining  = hi - lo;
        if (nRemaining < 2)
            return;  // Arrays of size 0 and 1 are always sorted

        // If array is small, do a "mini-TimSort" with no merges
        if (nRemaining < MIN_MERGE) {
            int initRunLen = countRunAndMakeAscending(a, lo, hi);
            binarySort(a, lo, hi, lo + initRunLen);
            return;
        }

        /**
         * March over the array once, left to right, finding natural runs,
         * extending short natural runs to minRun elements, and merging runs
         * to maintain stack invariant.
         */
        ComparableTimSort ts = new ComparableTimSort(a, work, workBase, workLen);
        int minRun = minRunLength(nRemaining);
        do {
            // Identify next run
            int runLen = countRunAndMakeAscending(a, lo, hi);

            // If run is short, extend to min(minRun, nRemaining)
            if (runLen < minRun) {
                int force = nRemaining <= minRun ? nRemaining : minRun;
                binarySort(a, lo, lo + force, lo + runLen);
                runLen = force;
            }

            // Push run onto pending-run stack, and maybe merge
            ts.pushRun(lo, runLen);
            ts.mergeCollapse();

            // Advance to find next run
            lo += runLen;
            nRemaining -= runLen;
        } while (nRemaining != 0);

        // Merge all remaining runs to complete sort
        assert lo == hi;
        ts.mergeForceCollapse();
        assert ts.stackSize == 1;
    }

line4的nRemaining表示沒有排序的對象個數，方法執行前，如果這個數小於2，就不需要排序了。

如果2<= nRemaining <=32,即MIN_MERGE的初始值，表示需要排序的數組是小數組，可以使用mini-TimSort方法進行排序，否則需要使用歸並排序。

mini-TimSort排序方法：先找出數組中從下標為0開始的第一個升序序列，或者找出降序序列后轉換為升序重新放入數組，將這段升序數組作為初始數組，將之后的每一個元素通過二分法排序插入到初始數組中。注意，這里就調用到了我們重寫的compareTo()方法了。

獲取初始數組的方法：

    private static int countRunAndMakeAscending(Object[] a, int lo, int hi) {
        assert lo < hi;
        int runHi = lo + 1;
        if (runHi == hi)
            return 1;

        // Find end of run, and reverse range if descending
        if (((Comparable) a[runHi++]).compareTo(a[lo]) < 0) { // Descending
            while (runHi < hi && ((Comparable) a[runHi]).compareTo(a[runHi - 1]) < 0)
                runHi++;
            reverseRange(a, lo, runHi);
        } else {                              // Ascending
            while (runHi < hi && ((Comparable) a[runHi]).compareTo(a[runHi - 1]) >= 0)
                runHi++;
        }

        return runHi - lo;
    }

根據程序中舉例，a[1].compareTo(a[0]) <0,所以向下循環查看a[2].compareTo(a[1]) <0、a[3].compareTo(a[2]) <0等等是否成立，我們發現a[2].compareTo(a[1]) <0不成立，所以循環終止，獲取到最長的降序數組為a[]{5,3},再調用reverseRange()方法將其升序排列為a[]{3,5},作為初始數組，initRunLen=2。隨后進行二分法插入操作，代碼如下：

private static void binarySort(Object[] a, int lo, int hi, int start) {
        assert lo <= start && start <= hi;
        if (start == lo)
            start++;
        for ( ; start < hi; start++) {
            Comparable pivot = (Comparable) a[start];

            // Set left (and right) to the index where a[start] (pivot) belongs
            int left = lo;
            int right = start;
            assert left <= right;
            /*
             * Invariants:
             *   pivot >= all in [lo, left).
             *   pivot <  all in [right, start).
             */
            while (left < right) {
                int mid = (left + right) >>> 1;
                if (pivot.compareTo(a[mid]) < 0)
                    right = mid;
                else
                    left = mid + 1;
            }
            assert left == right;

            /*
             * The invariants still hold: pivot >= all in [lo, left) and
             * pivot < all in [left, start), so pivot belongs at left.  Note
             * that if there are elements equal to pivot, left points to the
             * first slot after them -- that's why this sort is stable.
             * Slide elements over to make room for pivot.
             */
            int n = start - left;  // The number of elements to move
            // Switch is just an optimization for arraycopy in default case
            switch (n) {
                case 2:  a[left + 2] = a[left + 1];
                case 1:  a[left + 1] = a[left];
                         break;
                default: System.arraycopy(a, left, a, left + 1, n);
            }
            a[left] = pivot;
        }
    }

循環下標>=2的所有元素，通過二分法將其插入到初始數組中的適當位置，這樣，通過調用元素的compareTo()方法進行排序的功能實現完畢。

用Collections.sort(List list，Comparator c) 方法實現排序：

該方法傳入一個比較器，用於比較各元素的大小。該方法不需要元素實現Comparable接口，但需要一個實現Comparator接口的實現類來實例化一個比較器，注意，這里的Comparator是一個接口而非類。這里通常采用匿名內部類的方法。

Collections.sort(stuList, new Comparator<Student>() {
            @Override
            public int compare(Student stu1, Student stu2) {
                return (stu1.getAge() < stu2.getAge()) ? -1 : (stu1.getAge() == stu2.getAge() ? 0 : 1);
            }
        });

這種方法實現排序的方式與上述方法基本相同。

先調用Collections.sort()方法，傳入集合和比較器，sort()方法調用List的sort方法，傳入比較器。（同上step1）代碼如下：

    public static <T> void sort(List<T> list, Comparator<? super T> c) {
        list.sort(c);
    }

 

List中sort()方法調用Arrays.sort()方法，傳入數組和比較器。（同上step2）

default void sort(Comparator<? super E> c) {
    Object[] a = this.toArray();
    Arrays.sort(a, (Comparator) c);
    ListIterator<E> i = this.listIterator();
    for (Object e : a) {
        i.next();
        i.set((E) e);
    }
}

Arrays.sort方法調用TimSort.sort()方法，代碼如下：

public static <T> void sort(T[] a, Comparator<? super T> c) {
        if (c == null) {
            sort(a);
        } else {
            if (LegacyMergeSort.userRequested)
                legacyMergeSort(a, c);
            else
                TimSort.sort(a, 0, a.length, c, null, 0, 0);
        }
    }

legacyMergeSort(a,c)和TimSort.sort()方法中與方法一不同的地方只有一點，即方法一中使用a.compareTo(b)進行比較而方法二中使用comparator.compare(a,b)進行比較，其他均相同。

private static <T> int countRunAndMakeAscending(T[] a, int lo, int hi,
                                                    Comparator<? super T> c) {
        assert lo < hi;
        int runHi = lo + 1;
        if (runHi == hi)
            return 1;

        // Find end of run, and reverse range if descending
        if (c.compare(a[runHi++], a[lo]) < 0) { // Descending
            while (runHi < hi && c.compare(a[runHi], a[runHi - 1]) < 0)
                runHi++;
            reverseRange(a, lo, runHi);
        } else {                              // Ascending
            while (runHi < hi && c.compare(a[runHi], a[runHi - 1]) >= 0)
                runHi++;
        }

        return runHi - lo;
    }

總結：

1.Collections.sort()排序有兩種實現方式，一是讓元素類實現Comparable接口並覆蓋compareTo()方法，二是給Collecitons.sort()方法傳入比較器，通常采用匿名內部內的方式傳入。

2.Collections.sort()通過調用Arrays.sort()方法進行排序，在Java1.6+中，如果集合大小<32則采用Tim-Sort算法，如果>=32則采用歸並排序。