pta7-4 插入排序还是归并排序

根据维基百科的定义：

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

归并排序进行如下迭代操作：首先将原始序列看成 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
和之前插入排序还是堆排序思路基本上一样啦~
（忘记了的指路：https://editor.csdn.net/md/?articleId=125011221）
注意点大概就是还是判断的是<=…

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string>
#include<cstring>
#include<vector>
#include<set>
#include<map>
using namespace std;
#pragma warning(disable:4996)
const int N = 110;
int a[N], s[N], n;

int main() {
	cin >> n;
	for (int i = 0; i < n; i++) cin >> a[i];
	for (int i = 0; i < n; i++) cin >> s[i];
	int i, j;
	for (i = 0; i < n && s[i] <= s[i+1]; i++);//一定要写<=不然会段错误
	for (j = i + 1; j < n; j++) {
		if (a[j] != s[j])break;
	}
	if (j == n) {
		cout << "Insertion Sort" << endl;
		sort(a, a + min(i+2,n));
	}
	else {
		cout << "Merge Sort" << endl;
		int flag = 1, step = 1;
		while (flag) {
			flag = 0;
			for (int k = 0; k < n; k++) {
				//判断是否与给出的中间序列相同
				if (a[k] != s[k]) {
					flag = 1;
				}
			}
			step = step * 2;
			for (int k = 0; k < n; k+=step) {//模拟归并排序
				sort(a + k, a + min(k + step, n));//相同后还要再排序一次
			}
		}
	}
	for (int k = 0; k < n; k++) {
		if (k)cout << " " << a[k];
		else cout << a[k];
	}
}