PAT 1098 Insertion or Heap Sort (25 分)

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
void downadjust(vector<int> &b,int low,int high)
{
    int i=low,j=2*low;
    while(j<=high)
    {
        //右孩子比左孩子大，j指向右孩子
        if(j+1<=high&&b[j+1]>b[j])
            j=j+1;
        if(b[j]<=b[i])
            break;
        swap(b[j],b[i]);
        i=j;
        j=i*2;
    }
}
int main()
{
    int n,p=2;
    scanf("%d",&n);
    vector<int> a(n+1),b(n+1);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1;i<=n;i++)
        scanf("%d",&b[i]);
    //找到b的有序序列最后一个元素，从这之后开始比较a，b，若相同则为插入排序，反之为堆排序
    while(p<=n&&b[p-1]<=b[p])
        p++;
    int index=p;
    while(p<=n&&a[p]==b[p])
        p++;
    if(p==n+1)
    {
        //若为插入排序，将下一个元素和之前的有序序列一起排序
        printf("Insertion Sort\n");
        sort(b.begin()+1,b.begin()+index+1);
    }
    else
    {
        printf("Heap Sort\n");
        //从后往前找到第一个小于b[1]的数
        while(n>2&&b[n]>=b[1])
            n--;
        //两者交换
        swap(b[1],b[n]);
        //向下调整
        downadjust(b,1,n-1);
    }
    for(int i=1;i<b.size();i++)
        printf("%d%s",b[i],i==b.size()-1?"":" ");
    return 0;
}