pat 甲 1098 Insertion or Heap Sort （测试点4）

题目详情 - 1098 Insertion or Heap Sort (25 分) (pintia.cn)

判断插入排序或者堆排序，要注意测试点4的坑，可能存在重复的数据

#include<cstdio>
#include<cstdlib>
#include<algorithm>
using namespace std;
int n,array[110],ans[110];
void downadjust(int pos){
    swap(ans[1],ans[pos]);
    int low=1,height=pos-1;
    while(low<=height){
        int j=low*2;
        if(j>height) break;
        int t=j;
        if(j+1<=height&&ans[j+1]>ans[j]){
            t=j+1;
        }
        if(ans[low]<ans[t]){
            swap(ans[low],ans[t]);
            low=t;
        }
        else break;
    }
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&array[i]);
    }
    for(int i=1;i<=n;i++){
        scanf("%d",&ans[i]);
    }
    int judge=1;
    int pos=1;
    while(pos<n&&ans[pos]<=ans[pos+1]){ //此处要注意用 <= 否则测试点4无法AC
        pos++;
    }
    for(int i=pos+1;i<=n;i++){
        if(array[i]!=ans[i]){
            judge=-1;
            break;
        }
    }
    if(judge==1){
        printf("Insertion Sort\n");
        int temp=ans[pos+1],i;
        for(i=pos;i>=1;i--){
            if(ans[i]>temp){
                ans[i+1]=ans[i];
            }
            else{
                ans[i+1]=temp;
                break;
            }
        }
        if(i==0) ans[1]=temp;
    }
    else{
        printf("Heap Sort\n");
        int temp=ans[1];
        pos=n;
        while(pos>1&&ans[pos]>temp){
            pos--;
        }
        downadjust(pos);
    }
    printf("%d",ans[1]);
    for(int i=2;i<=n;i++){
        printf(" %d",ans[i]);
    }
    system("pause");
    return 0;
}

测试点4的数据类似于

10

3 5 1 5 4 2 7 8 6 9

1 3 4 5 5 2 7 8 6 9

此数据若判断代码中用 < 会误判为堆排序

或者

10

3 5 1 4 5 2 7 8 6 9

1 3 4 5 5 2 7 8 6 9

此数据判断代码中用 < 和用 <= 均正确判断为插入排序，但会产生不同的输出序列