算法笔记练习 4.6 two pointers 问题 A: 二路归并排序(mergesort)递归法

算法笔记练习 题解合集

题目链接

题目

题目描述
二路归并排序(mergesort)递归法，用递归法进行二路归并排序

输入：
第一行一个数据n，表示有n个数要排序。接下来n行每行一个<=10^7的整数。

输出：
n行，由小到大排序后的数据
数据规模：n<=10^5

思考：两个递归都会被执行吗？

思路

思路参考算法笔记 P140，我按自己的习惯实现了同样的功能。

代码

#include <stdio.h>

// 把数组 a 中分别递增的两半合并 
void merge(int *a, int L1, int R1, int L2, int R2) {
	int temp[R2 - L1 + 1];
	int *ptemp = temp;
	int *pa = a + L1;
	int *pb = a + L2;
	while(pa <= a + R1 && pb <= a + R2) {
		if (*pa < *pb)
			*ptemp++ = *pa++;
		else
			*ptemp++ = *pb++;
	}
	while (pa <= a + R1)
		*ptemp++ = *pa++;
	while (pb <= a + R2)
		*ptemp++ = *pb++;
	int i;
	for (i = 0; i < R2 - L1 + 1; ++i)
		a[L1 + i] = temp[i];
} 

// 二路归并排序，递归写法 
void mergeSort(int n, int *a) {
	if (n == 0 || n == 1)
		return;
	mergeSort(n / 2, a);
	mergeSort(n - n / 2, a + n / 2);
	merge(a, 0, n / 2 - 1, n / 2, n - 1);
}

int main() {
	int n;
	while (scanf("%d", &n) != EOF) {
		int nums[n], i;
		for (i = 0; i < n; ++i)
			scanf("%d", &nums[i]);
		mergeSort(n, nums);
		for (i = 0; i < n; ++i)
			printf("%d\n", nums[i]);
	} 
	return 0;
}