【PAT B1035/A1089】 插入与归并

根据维基百科的定义：

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

归并排序进行如下迭代操作：首先将原始序列看成 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 <cstdio>
#include <algorithm>

using namespace std;

int main() {
    int n, a[110], b[110], i, j;
    scanf("%d", &n);
    for (i = 0; i < n; i++) {
        scanf("%d", &a[i]);
    }
    for (i = 0; i < n; i++) {
        scanf("%d", &b[i]);
    }
    for (i = 0; i < n - 1 && b[i] <= b[i + 1]; i++);//找到第一个未排序的位置
    for (j = i + 1; a[j] == b[j] && j < n; j++);//从这个位置开始匹配a和b数组
    if (j == n) {//剩余的元素都没有移动
        printf("Insertion Sort\n");
        sort(a, a + i + 2);
    } else {
        printf("Merge Sort\n");
        int k = 1, flag = 1;
        while (flag) {//flag为1时继续加倍k
            flag = 0;
            for (i = 0; i < n; i++) {//对比a、b，当a和b完全相同时flag为0
                if (a[i] != b[i]) {
                    flag = 1;
                }
            }
            k *= 2;//进行下一步排序
            for (i = 0; i < n / k; i++) {//排序前面等分的部分
                sort(a + i * k, a + (i + 1) * k);
            }
            sort(a + i * k, a + n);//排序剩余部分
        }
    }
    for (i = 0; i < n; i++) {
        if (i != 0) printf(" ");
        printf("%d", a[i]);
    }
    return 0;
}