【PAT乙级】1035 插入与归并 (25分)

根据维基百科的定义：

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

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

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

输入格式：

输入在第一行给出正整数 N ($\le$ 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

输入样例1：

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

解题思路：

// 插入排序模板
for (int i = 1; i < N; i++) {                           
	for (int j = i-1; j >= 0 && a[j] > a[j+1]; j--) {
		swap(a[j], a[j+1]);
	}
}

// 归并排序模板
for (int l = 2; l/2 <= N; l += l) {                     
	for (int i = 0; i < N; i += l) {
		sort(a.begin()+i, a.begin()+min(i+l, N));
	}
}

答案代码：

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int N = 0;
bool insert_sort(vector<int> a, vector<int> &b) {
	bool flag = false;
	for (int i = 1; i < N; i++) {                           // 插入排序模板
		for (int j = i-1; j >= 0 && a[j] > a[j+1]; j--) {
			swap(a[j], a[j+1]);
		}
		if (flag) { b = a; return true; }					// flag = true 上一轮相同
		if (a == b) flag = true;
	}
    return false;
}

bool merge_sort(vector<int> a, vector<int> &b) {
	bool flag = false;
	for (int l = 2; l/2 <= N; l += l) {                     // 归并排序模板
		for (int i = 0; i < N; i += l) {
			sort(a.begin()+i, a.begin()+min(i+l, N));
		}
		if (flag) { b = a; return true; }					// flag = true 上一轮相同
		if (a == b) flag = true;
	}
	return false;
}

int main() {
	cin >> N;
	
	vector<int> a(N), b(N);
	for (int i = 0; i < N; i++)
		cin >> a[i];
	for (int i = 0; i < N; i++)
		cin >> b[i];
	
	if (merge_sort(a,b)) cout << "Merge Sort" << endl;
	else if (insert_sort(a,b)) cout << "Insertion Sort" << endl;

	for (int i = 0; i < b.size(); i++) {
		cout << b[i];
		if (i != b.size()-1) cout << " ";
	}
	return 0;
}