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中的排序函数，而是自己写了插入和归并的算法。
感谢该作者。
刚刚去看了一下这个作者的介绍，发现是个大三的学生。
有点羞愧。果然四年逃的课要用更长的时间补回来啊。