PAT 1089 Insert or Merge (25 分)

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
int main()
{
    int n;
    scanf("%d",&n);
    vector<int> a(n),b(n);
    for(int i=0;i<n;i++)
        scanf("%d",&a[i]);
    for(int i=0;i<n;i++)
        scanf("%d",&b[i]);
    int i,j;
    //用i标注有序的长度
    for(i=0;i<n-1&&b[i]<=b[i+1];i++);
    //在有序序列之后，a，b相同，则为插入排序，否则为归并排序
    for(j=1+i;a[j]==b[j]&&j<n;j++);
    if(j==n)
    {
        printf("Insertion Sort\n");
        sort(a.begin(),a.begin()+i+2);
    }
    else
    {
        printf("Merge Sort\n");
        int k=1,flag=1;
        while(flag)
        {
            flag=0;
            //先将a归并排序成和b相同，在此期间，flag为1，相同之后，flag为0，进行最后一次排序
            for(i=0;i<n;i++)
            {
                if(a[i]!=b[i])
                    flag=1;
            }
            k*=2;
            //归并排序
            for(i=0;i<n/k;i++)
                sort(a.begin()+i*k,a.begin()+(i+1)*k);
            //最后不到k个元素，排序
            sort(a.begin()+n/k*k,a.end());
        }
    }
    for(j=0;j<n;j++)
    {
        if(j!=0)
            printf(" ");
        printf("%d",a[j]);
    }
    return 0;
}