算法设计与分析之归并排序

简介

归并排序是建立在归并操作上的一种有效，稳定的排序算法，该算法是采用分治法（Divide and Conquer）的一个非常典型的应用。将已有序的子序列合并，得到完全有序的序列；即先使每个子序列有序，再使子序列段间有序。若将两个有序表合并成一个有序表，称为二路归并。

算法描述

归并操作的工作原理如下：
第一步：申请空间，使其大小为两个已经排序序列之和，该空间用来存放合并后的序列
第二步：设定两个指针，最初位置分别为两个已经排序序列的起始位置
第三步：比较两个指针所指向的元素，选择相对小的元素放入到合并空间，并移动指针到下一位置
重复步骤3直到某一指针超出序列尾
将另一序列剩下的所有元素直接复制到合并序列尾

代码示例

#include<iostream>
#include<cstdlib>
using namespace std;
void mergeSort(int* A, int lenA);
void merge(int* B, int lenB, int* C, int lenC, int* A, int lenA);
int main() {
	int n;
	cout << "请输入数组长度：";
	cin >> n;
	cout << "\n输入整数数组：";
	int* a = (int*)malloc(sizeof(int) * n);		// 动态分配
	for (int i = 0; i < n; i++) {
		cin >> a[i];
	}
	mergeSort(a, n);
	cout << "\n排序后数组为：";
	for (int i = 0; i < n; i++) {
		cout << a[i] << " ";
	}
}
void mergeSort(int* A, int lenA) {
	if (lenA > 1) {	// 数组A还能被分配
		// 分治
		int n1 = lenA / 2;
		int n2 = lenA - n1;
		// 将A拆成B与C两个数组
		int* B = (int*)malloc(sizeof(int) * n1);	// 动态分配
		int* C = (int*)malloc(sizeof(int) * n2);
		for (int i = 0; i < n1; i++) {
			B[i] = A[i];
		}
		for (int i = 0; i < n2; i++) {
			C[i] = A[n1 + i];
		}
		// 递归
		mergeSort(B, n1);
		mergeSort(C, n2);
		// 合并
		merge(B, n1, C, n2, A, lenA);
		// 释放内存
		free(B);
		free(C);
	}
}
void merge(int* B, int lenB, int* C, int lenC, int* A, int lenA) {	// 合并
	int i = 0, j = 0, k = 0;		// i对应数组B, j对应数组C, k对应数组A
	while (i < lenB && j < lenC) {	// 数组B和数组C的数都没被分配完的情况
		if (B[i] <= C[j]) 
			A[k++] = B[i++];	// 升序，从小到大	 将较小的先合并进新数组
		else A[k++] = C[j++];
	}
	if (i == lenB) {	// 数组B被分配完，数组C剩余的依次分配进新数组
		while (j < lenC) 
			A[k++] = C[j++];
	}
	else {				// j == lenC	数组C被分配完，数组B剩余的依次分配进新数组
		while (i < lenB)
			A[k++] = B[i++];
	}
}