數據結構-歸併排序

java實現歸併排序算法

源代碼

public class MergeSort extends DataCrol {
    // 合併兩個已排好序的數組A[left...mid]和A[mid+1...right]
    void merge(int A[], int left, int mid, int right) {
        int len = right - left + 1;
        int[] temp = new int[len];// 輔助空間O(n)
        int index = 0;
        int i = left;                   // 前一數組的起始元素
        int j = mid + 1;                // 後一數組的起始元素
        while (i <= mid && j <= right) {
            temp[index++] = A[i] <= A[j] ? A[i++] : A[j++];  // 帶等號保證歸併排序的穩定性
        }
        while (i <= mid) {
            temp[index++] = A[i++];
        }
        while (j <= right) {
            temp[index++] = A[j++];
        }
        for (int k = 0; k < len; k++) {
            A[left++] = temp[k];
        }
    }

    // 遞歸實現的歸併排序(自頂向下)
    void mergeSortRecursion(int A[], int left, int right) {
        if (left == right)    // 當待排序的序列長度爲1時，遞歸開始回溯，進行merge操作
            return;
        int mid = (left + right) / 2;
        mergeSortRecursion(A, left, mid);
        mergeSortRecursion(A, mid + 1, right);
        merge(A, left, mid, right);
    }

    // 非遞歸(迭代)實現的歸併排序(自底向上)
    void mergeSortIteration(int A[]) {
        int len = A.length;
        int left, mid, right;// 子數組索引,前一個爲A[left...mid]，後一個子數組爲A[mid+1...right]
        for (int i = 1; i < len; i *= 2) {       // 子數組的大小i初始爲1，每輪翻倍
            left = 0;
            while (left + i < len) {              // 後一個子數組存在(需要歸併)
                mid = left + i - 1;
                right = mid + i < len ? mid + i : len - 1;// 後一個子數組大小可能不夠
                merge(A, left, mid, right);
                left = right + 1;               // 前一個子數組索引向後移動
            }
        }
    }

    @Override
    public void sort(int[] array) {
//        mergeSortIteration(array);
        mergeSortRecursion(array, 0, array.length - 1);
    }

    public static void main(String[] args) {
        MergeSort mergeSort = new MergeSort();
        int[] array = DataCrol.createRandomArray(20);
        DataCrol.print(array);
        mergeSort.sort(array);
        DataCrol.print(array);
        mergeSort.timeTest(10000000);
    }
}