PAT 堆——A1098.Insertion or Heap Sort(25)（内涵堆的详细创建过程）

#include <bits/stdc++.h>
#include <stdio.h>
#include <stdlib.h>
#include <queue>
using namespace std;
const int N = 111;
int origin[N],tempOri[N],changed[N];//原始数组,原始数组备份及目标数组
int n;//元素个数
bool isSame(int A[],int B[]){
    for(int i=1;i<=n;++i){
        if(A[i] != B[i]){
            return false;
        }
    }
    return true;
}
bool showArray(int A[]){
    for(int i=1;i<=n;++i){
        printf("%d",A[i]);
        if(i<n){
            printf(" ");
        }
    }
}
bool insertSort(){//插入排序
    bool flag =false;
    for(int i=2;i<=n;++i){
        if(i != 2 && isSame(tempOri,changed)){
            flag = true;
        }
        //插入部分直接用sort代替
        sort(tempOri,tempOri + i + 1);
        if(flag == true){
            return true;
        }
    }
    return false;
}

//对heap数组在[low,high]范围进行调整
void downAdjust(int low,int high){
    int i = low,j = i * 2;
    while(j <= high){//存在孩子节点
        if(j + 1 <= high && tempOri[j+1] > tempOri[j]){
            j = j + 1;
        }
        if(tempOri[j] > tempOri[i]){
            swap(tempOri[j],tempOri[i]);
            i = j;
            j = i * 2;
        }else{
            break;
        }
    }
}
void heapSort(){
    bool flag = false;
    for(int i=n /2;i>=1;i--){//建堆
        downAdjust(i,n);
    }
    for(int i=n;i>1;--i){
        if(i != n && isSame(tempOri,changed)){
            flag = true;
        }
        swap(tempOri[i],tempOri[1]);
        downAdjust(1,i-1);//调整堆顶
        if(flag == true){
            showArray(tempOri);
            return;
        }
    }
    
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;++i){
        scanf("%d",&origin[i]);
        tempOri[i] = origin[i];
    }
    for(int i=1;i<=n;++i){
        scanf("%d",&changed[i]);
    }
    if(insertSort()){
        printf("Insertion Sort\n");
        showArray(tempOri);
    }else{
        printf("Heap Sort\n");
        for(int i=1;i<=n;++i){//还原tempOri数组
            tempOri[i] = origin[i];
        }
        heapSort();
    }
    system("pause");
    return 0;
}