[分治]-归并排序及其优化

一、概述

归并排序算法是用分治策略实现对n个元素进行排序的算法。基本思想是将待排序元素分成大小大致相同的两个子集合，分别对两个子集合进行排序，最终将排好序的子集合合并成锁要求的排好序的集合。通俗来讲就是对集合进行不断的分割，分割到只剩一个元素，最后进行比较，再两两合并到新集合中。
一个简单的例子示意如下：

二、递归版代码实现

1、递归排序

void MergeSort(int *a, int left, int right){
    int len = right-left+1;
    int b[len];
    if(left < right){
        int mid = (left + right)/2;
        MergeSort(a, left, mid); // 归并排序左端
        MergeSort(a, mid+1, right); // 归并排序右端
        Merge(a, b, left, mid, right); // 合并好两个排好序的数组到数组b
        Copy(a, b, left, right); // 将合并好的b数组再复制到数组a中
    }
}

2、合并函数

// 归并排序好的数组两端合成新数组
void Merge(int *a, int *b, int left, int mid, int right){
	int i, j, cnt; // 定义i，j分别迭代遍历左端和右端，并比较左右端将小的放前边
    for(i = left, j = mid+1, cnt = left;i <= mid && j <= right; cnt++){
        if(a[i] < a[j]){
            b[cnt] = a[i++];
        }else{
            b[cnt] = a[j++];
        }
    }

    while(i <= mid)
        b[cnt++] = a[i++]; // 若左端子集还有剩余
    while(j <= right)
        b[cnt++] = a[j++]; // 若右端子集还有剩余
}

3、数组拷贝

// 数组拷贝
void Copy(int *a, int *b,int left, int right){
    for(int i = left; i <= right; i++){
        a[i] = b[i];
    }
}

三、时间复杂度

Merge和Copy函数显然可以在O(n)时间内完成，时间大部分用在了递归函数上，因此合并排序算法对n个元素进行排序，在最坏情况下所需的计算时间T(n)满足：

解此递归方程可得
$$T(n)=O(n\log n)$$

四、优化

对于MergeSort，还可以从多方面对它进行改进。例如，从分治策略的机制入手，容易消除算法的递归。事实上，算法MergeSort的递归过程知识将待排序集合一分为二，直至待排序集合只剩一个元素为止，然后不断合并两个排好序的数组段。按此机制，可以首先将数组a中相邻元素两两配对。用合并算法将他们排序，构成n/2组长度为2的排好序的子数组段，然后再将它们排序成长度为4的排好序的子数组段，如此持续下去，直到整个数组排好序。

void Merge(int *c, int *d, int left, int mid, int right){
    // 合并c[从left到m]和c[从mid+1到right] 至->d[从1到right]
    int i = left, j = mid + 1, cnt = left;
    while(i <= mid && j <= right){
        if(c[i] <= c[j])
            d[cnt++] = c[i++];
        else
            d[cnt++] = c[j++];
    }

    while(i <= mid)
        d[cnt++] = c[i++]; // 若左端子集还有剩余
    while(j <= right)
        d[cnt++] = c[j++]; // 若右端子集还有剩余
}


void MergePass(int *a, int *b, int s, int n){
    // 合并大小为s的相邻子数组
    int left = 0;
    while(left <= n-2*s){
        // 合并大小为s的相邻两段子数组
        Merge(a,b,left,left+s-1,left+2*s-1);
        left += 2*s; // 右移2s
    }

    // 剩余元素少于2s
    if(left + s < n) Merge(a,b,left,left+s-1,n-1);
    else
        for(int j = left;j <= n-1;j++)
            b[j] = a[j];
}
void MergeSort(int *a, int n){
    int *b = new int [n];
    int s = 1;
    while(s < n){
        MergePass(a,b,s,n); // 合并到数组b
        s += s; // s扩大两倍
        MergePass(b,a,s,n); // 合并到数组a
        s += s;
    }
}