【PAT甲级】1098 Insertion or Heap Sort (25 分)

题意：

输入一个正整数N（<=100），接着输入两行N个数，表示原数组和经过一定次数排序后的数组。判断是经过插入排序还是堆排序并输出再次经过该排序后的数组（数据保证答案唯一）。

AAAAAccepted code：

#define HAVE_STRUCT_TIMESPEC
#include<bits/stdc++.h>
using namespace std;
int a[107],b[107];
int c[107][5];
int main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    int n;
    cin>>n;
    for(int i=1;i<=n;++i)
        cin>>a[i];
    for(int i=1;i<=n;++i)
        cin>>b[i];
    int pos=0;
    for(int i=1;i<n;++i)
        if(b[i]>b[i+1]){
            pos=i;
            break;
        }
    int pos2=0;
    for(int i=pos+1;i<=n;++i)
        if(a[i]!=b[i]){
            pos2=i;
            break;
        }
    if(!pos2){
        cout<<"Insertion Sort
";
        int flag=0,mn=b[pos+1];
        int ppos=0;
        for(int i=1;i<=n;++i)
            if(b[i]>mn){
                ppos=i;
                break;
            }
        for(int i=pos+1;i>ppos;--i)
            b[i]=b[i-1];
        b[ppos]=mn;
        for(int i=1;i<=n;++i){
            cout<<b[i];
            if(i<n)
                cout<<" ";
        }
    }
    else{
        cout<<"Heap Sort
";
        int ppos=0;
        sort(a+1,a+1+n);
        for(int i=n;i;--i)
            if(b[i]!=a[i]){
                ppos=i;
                break;
            }
        int mx=0,flag=0;
        for(int i=1;i<=ppos;++i)
            if(b[i]>mx){
                mx=b[i];
                flag=i;
            }
        swap(b[flag],b[ppos]);
        int s=1;
        while(s<ppos){
            if(b[s*2]>b[s]&&b[s*2+1]>b[s]&&ppos>s*2+1){
                if(b[s*2]>b[s*2+1]){
                    swap(b[s],b[s*2]);
                    s=s*2;
                }
                else{
                    swap(b[s],b[s*2+1]);
                    s=s*2+1;
                }
            }
            else if(b[s*2+1]>b[s]&&ppos>s*2+1){
                swap(b[s],b[s*2+1]);
                s=s*2+1;
            }
            else if(b[s*2]>b[s]&&ppos>s*2){
                swap(b[s],b[s*2]);
                s=s*2;
            }
            if(s*2>=ppos)
                break;
        }
        for(int i=1;i<=n;++i){
            cout<<b[i];
            if(i<n)
                cout<<" ";
        }
    }
    return 0;
}