【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[],b[];
 int c[][];
 int main(){
     ios::sync_with_stdio(false);
     cin.tie(NULL);
     cout.tie(NULL);
     int n;
     cin>>n;
     for(int i=;i<=n;++i)
         cin>>a[i];
     for(int i=;i<=n;++i)
         cin>>b[i];
     int pos=;
     for(int i=;i<n;++i)
         if(b[i]>b[i+]){
             pos=i;
             break;
         }
     int pos2=;
     for(int i=pos+;i<=n;++i)
         if(a[i]!=b[i]){
             pos2=i;
             break;
         }
     if(!pos2){
         cout<<"Insertion Sort\n";
         int flag=,mn=b[pos+];
         int ppos=;
         for(int i=;i<=n;++i)
             if(b[i]>mn){
                 ppos=i;
                 break;
             }
         for(int i=pos+;i>ppos;--i)
             b[i]=b[i-];
         b[ppos]=mn;
         for(int i=;i<=n;++i){
             cout<<b[i];
             if(i<n)
                 cout<<" ";
         }
     }
     else{
         cout<<"Heap Sort\n";
         int ppos=;
         sort(a+,a++n);
         for(int i=n;i;--i)
             if(b[i]!=a[i]){
                 ppos=i;
                 break;
             }
         int mx=,flag=;
         for(int i=;i<=ppos;++i)
             if(b[i]>mx){
                 mx=b[i];
                 flag=i;
             }
         swap(b[flag],b[ppos]);
         int s=;
         while(s<ppos){
             if(b[s*]>b[s]&&b[s*+]>b[s]&&ppos>s*+){
                 if(b[s*]>b[s*+]){
                     swap(b[s],b[s*]);
                     s=s*;
                 }
                 else{
                     swap(b[s],b[s*+]);
                     s=s*+;
                 }
             }
             else if(b[s*+]>b[s]&&ppos>s*+){
                 swap(b[s],b[s*+]);
                 s=s*+;
             }
             else if(b[s*]>b[s]&&ppos>s*){
                 swap(b[s],b[s*]);
                 s=s*;
             }
             if(s*>=ppos)
                 break;
         }
         for(int i=;i<=n;++i){
             cout<<b[i];
             if(i<n)
                 cout<<" ";
         }
     }
     return ;
 }