PAT A1098 insertion or heap sort (25)

和A1089类似

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
const int MAX=105;
int n;
int init[MAX],parts[MAX],nexts[MAX],sorted[MAX];


int cmp(int a,int b){
return a<b;
}
int judge(){
int i,j,tag=0;
int temp;
for(i=2;i<=n;i++){//find i pos 
tag=0;
for(j=i;j>1;j--){
if(sorted[j]<sorted[j-1]){
temp=sorted[j];
sorted[j]=sorted[j-1];
sorted[j-1]=temp;
}else break;
}
for(j=1;j<=n;j++){
if(sorted[j]!=parts[j]){
tag=1;
break;
}
}
if(tag==0) return i;
}
return -1;


}








void insertsort(int i){
int temp;
for(int j=i-1;j>=1;j--){
if(parts[j]>parts[j+1]){
temp=parts[j];
parts[j]=parts[j+1];
parts[j+1]=temp;
}else break;
}
for(int j=1;j<=n;j++){
if(j<n) printf("%d ",parts[j]);
else printf("%d\n",parts[j]);
}
}
int heap[102];
void downadjust(int low,int high){
int i=low;
int j=low*2;
while(j<=high){
if((j+1)<=high &&init[j+1]>init[j]){
j=j+1;
}
if(init[i]<init[j]){
swap(init[i],init[j]);
i=j;
j=i*2;
}else break;

}
}


int issame(){
int i,tag=0;
for(i=1;i<=n;i++){
if(init[i]!=parts[i]){
tag=1;
break;
}
}
if(tag==1) return -1;
else return 1;
} 
void heapsort(){

int i;
for(i=(n/2);i>=1;i--){
downadjust(i,n);
}
for(i=n;i>1;i--){
swap(init[1],init[i]);
downadjust(1,i-1);
if(issame()==1) break;
}
i--; 
swap(init[1],init[i]);
downadjust(1,i-1);
for(int j=1;j<=n;j++){
if(j<n) printf("%d ",init[j]);
else printf("%d\n",init[j]);
}
}


int main(){
scanf("%d",&n);
int i,temp;
for(i=1;i<=n;i++){
scanf("%d",&temp);
init[i]=temp;
sorted[i]=temp;
}
for(i=1;i<=n;i++){
scanf("%d",&temp);
parts[i]=temp;
}
int result=judge();
if(result!=-1){
printf("Insertion Sort\n");
insertsort(result+1);
}else{
printf("Heap Sort\n");
heapsort();
}
system("pause");
return 0;
}