归并排序实现

2019独角兽企业重金招聘Python工程师标准>>>

归并排序实现：

先来看一个数组：

int[] arr = new int[]{12, 23, 34, 11, 22, 44};

我们将其分为两个数组：

// 注意这两个数组都是有序的
leftArr:  [12, 23, 34]
rightArr: [11, 22, 44]

通过合并操作我们将其合成一个有序的数组，类似链表的合并，实现代码：

/**
* 对数组arr的两个左右子数组arr[p..q-1], arr[q..r]进行合并排序
* @param arr 待合并排序的数组
* @param p 左子数组的起始索引, 从0开始
* @param q 两个子数组分割索引
* @param r 第二个子数组的结束索引, 最大为arr.length-1
*/
private static void merge(int[] arr, int p, int q, int r) {
	int[] leftArr = new int[q - p + 1];
	int[] rightArr = new int[r - q];
		
	// 初始化左边的子数组
	for (int i=0; i<leftArr.length; i++){
	    leftArr[i] = arr[p+i];
	}
	// 初始化右边的子数组
	for (int i=0; i<rightArr.length; i++){
	    rightArr[i] = arr[q+i+1];
	}
		
	int leftIndex=0, rightIndex=0;
	int arrIndex = p;
	while (arrIndex < r){
		if (leftArr[leftIndex] <= rightArr[rightIndex]){
			arr[arrIndex++] = leftArr[leftIndex++];
		} else {
			arr[arrIndex++] = rightArr[rightIndex++];
		}
			
		// 若某子数组已经到数组尾部了
		if (leftIndex == leftArr.length 
				|| rightIndex == rightArr.length){
			break;
		}
	}
	// 如果左边子数组还有剩余, 接上
	if (leftIndex < leftArr.length){
		while(leftIndex < leftArr.length){
			arr[arrIndex++] = leftArr[leftIndex++];
		}
	}
	// 如果右边子数组还有剩余, 接上
	if (rightIndex < rightArr.length){
		while(rightIndex < rightArr.length){
			arr[arrIndex++] = rightArr[rightIndex++];
		}
	}
}

理解这步，就行了，那么要是对于乱序的数组怎么用合并呢，如：

int[] arr = new int[]{31, 21, 8, 11, 5, 1};

我们可以将数组不断细分，直至分成2个一组就是有序的了，这就是算法的分治思想：

/**
 * 归并排序
 * @param arr 待排序数组
 * @param i 数组起始索引
 * @param j 数组结束索引
 */
private static void mergeSort(int[] arr, int i, int j) {
	if (i < j){ 
		int k = (i+j) / 2;
		// 归并左边数组
		mergeSort(arr, i, k);
		// 归并右边数组
		mergeSort(arr, k+1, j);
		// 将两个[有序]数组合并
		merge(arr, i, k, j);
	}
}

测试用例：

int[] arr = new int[]{31, 21, 8, 11, 5, 1};
mergeSort(arr, 0, arr.length - 1);

其时间复杂度为nlgn.

转载于:https://my.oschina.net/indestiny/blog/201946