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

我的个人网站 Cheese的个人主页http://www.cheese.ren/

博客来源 PAT 乙级 1035 插入与归并-Cheese的个人博客http://blog.cheese.ren/55

欢迎交换友链 :-)

---

#include <bits/stdc++.h>

using namespace std;

int main() {
    int n, a[101], b[101];
    scanf("%d", &n);
    for (int i=0; i<n; i++) {
        scanf("%d", &a[i]);
    }
    for (int i=0; i<n; i++) {
        scanf("%d", &b[i]);
    }
    int temp = 0;// 搜索b[i]>b[i+1]的点
    for (int i=0; i<n; i++) {
        if (b[i] > b[i+1]) {
            temp = i;
            break;
        }
    }
    int flag = 0;// 1 -> Merge Sort, 0 -> Insertion Sort
    for (int i=temp+1; i<n; i++) {
        if (a[i] != b[i]) {
            flag = 1;
            break;
        }
    }
    if (flag) {
        printf("Merge Sort\n");
        //  a重新排序，直到排序排到b
        int k = 1;
        while (flag) {
            flag = 0;
            for (int i=0; i<n; i++) {
                if (a[i] != b[i]) {
                    flag = 1;
                    break;
                }
            }
            k *= 2;
            for (int i=0; i<n/k; i++) {
                sort(a+i*k, a+(i+1)*k);
            }
            //  注意余下的数
            sort(a+n/k*k, a+n);
        }
    }
    else {
        printf("Insertion Sort\n");
        sort(a, a+temp+2);
    }
    for (int i=0; i<n-1; i++) {
        printf("%d ", a[i]);
    }
    printf("%d\n", a[n-1]);
    return 0;
}