PAT 1035 插入与归并

根据维基百科的定义：

插入排序是迭代算法，逐一获得输入数据，逐步产生有序的输出序列。每步迭代中，算法从输入序列中取出一元素，将之插入有序序列中正确的位置。如此迭代直到全部元素有序。

归并排序进行如下迭代操作：首先将原始序列看成 N 个只包含 1 个元素的有序子序列，然后每次迭代归并两个相邻的有序子序列，直到最后只剩下 1 个有序的序列。

现给定原始序列和由某排序算法产生的中间序列，请你判断该算法究竟是哪种排序算法？

输入格式：

输入在第一行给出正整数 N (≤100)；随后一行给出原始序列的 N 个整数；最后一行给出由某排序算法产生的中间序列。这里假设排序的目标序列是升序。数字间以空格分隔。

输出格式：

首先在第 1 行中输出Insertion Sort表示插入排序、或Merge Sort表示归并排序；然后在第 2 行中输出用该排序算法再迭代一轮的结果序列。题目保证每组测试的结果是唯一的。数字间以空格分隔，且行首尾不得有多余空格。

输入样例 1：

10
3 1 2 8 7 5 9 4 6 0
1 2 3 7 8 5 9 4 6 0

输出样例 1：

Insertion Sort
1 2 3 5 7 8 9 4 6 0

输入样例 2：

10
3 1 2 8 7 5 9 4 0 6
1 3 2 8 5 7 4 9 0 6

输出样例 2：

Merge Sort
1 2 3 8 4 5 7 9 0 6

代码：

#include<iostream>
#include<algorithm>
const int maxn = 100 + 1;
using namespace std;
bool cmp(int a, int b)
{
	return a < b;
}
int main()
{
	int a[maxn];
	int b[maxn];
	int n;
	cin >> n;
	for (int i = 0; i < n; i++)
		cin >> a[i];
	for (int i = 0; i < n; i++)
		cin >> b[i];
	int i,j;
	for (i = 0; i < n - 1 && b[i] <= b[i + 1]; i++);
	for (j = i + 1; a[j] == b[j]&&j<n; j++);
	if (j == n)
	{
		cout << "Insertion Sort" << endl;
		sort(a, a+ i + 2, cmp);
	}
	else
	{
		cout << "Merge Sort" << endl;
		int k = 1;
		int flag = 1;
		while (flag)
		{
			flag = 0;
			for (int i = 0; i < n; i++)
			{
				if (a[i] != b[i])
					flag = 1;
			}
			k = k * 2;
			for (int i = 0; i < n / k; i++)
				sort(a + i*k, a + (i + 1)*k, cmp);
			sort(a + n / k*k, a + n, cmp);
		}
	}
	for (j = 0; j < n - 1; j++)
		cout << a[j] << " ";
	cout << a[n - 1];
	return 0;
}