PAT BASIC LEVEL 1035. 插入与归并(25)

1035. 插入与归并(25)

根据维基百科的定义：
插入排序
是迭代算法，逐一获得输入数据，逐步产生有序的输出序列。每步迭代中，算法从输入序列中取出一元素，将之插入有序序列中正确的位置。如此迭代直到全部元素有序。
归并排序
进行如下迭代操作：首先将原始序列看成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

Answer:

#include<iostream>
using namespace std;

bool equal_seq(int *a, int *b, int num) {
    for(int i = 0; i < num; i ++)
        if(a[i] != b[i])
            return false;
    return true;
}

void print_seq(int *seq, int num) {
    cout << seq[0];
    for(int i = 1;i < num;i ++)
        cout << ' ' << seq[i];
    cout << '\n';
}

void insertion_sort(int *res, int i, int n) {
    int t = res[i + 1];
    while(i >= 0 && res[i] > t)
        res[i+1] = res[i--];
    res[i+1] = t;
}

void merge_array(int *a, int *b, int s, int e, int m) {
    int i = s, j = m + 1;
    int k = 0;
    while(i <= m && j <= e)
        if(a[i] < a[j])
            b[k++] = a[i++];
        else
            b[k++] = a[j++];

    while(i <= m)
        b[k++] = a[i++];
    while(j <= e)
        b[k++] = a[j++];

    for(k--;k >= 0;k--)
        a[s + k] = b[k];
}

void merge_sort(int *seq, int *temp, int n, int *bak) {
    for(int step = 1;step < n;step *= 2) {
        int s = 0;
        int end = 2*step > n-1 ? n-1 : 2*step-1;
        while(s < n-step) {
            merge_array(seq, temp, s, end, s+step-1);
            s += 2*step;
            end = s + 2*step > n-1 ? n-1 : s+2*step-1;
        }
        if(!equal_seq(seq, bak, n))
            return;
    }
}


int main() {
    int n;
    cin >> n;
    int seq[n], res[n];
    for(int i = 0;i < n;i ++)
        cin >> seq[i];
    for(int i = 0;i < n;i ++)
        cin >> res[i];
    int i, j;
    for (i = 0; i < n - 1 && res[i] <= res[i + 1]; i++);
    for (j = i + 1; seq[j] == res[j] && j < n; j++);
    if(j == n) {
        cout << "Insertion Sort\n";
        insertion_sort(res, i, n);
    } else {
        int t[n], bak[n];
        for(int i = 0;i < n;i ++)
            bak[i] = res[i];
        merge_sort(res, t, n, bak);
        cout << "Merge Sort\n";
    }
    print_seq(res, n);
}

PS.
这道题啊……
第三个测试一直不通过，本来以为是归并排序有问题（毕竟归并排序比插入排序要难一点），检查了好久好久好久。
然而错的是插入排序。
所以改了一下判断的算法，参考了这篇博客。不过因为某种奇怪的原因，没有使用algorithm中的排序函数，而是自己写了插入和归并的算法。
感谢该作者。
刚刚去看了一下这个作者的介绍，发现是个大三的学生。
有点羞愧。果然四年逃的课要用更长的时间补回来啊。