题目描述
知识点: 堆排序,插入排序
思路: 跟前面有一题很像,没啥好说的,先判断是不是插入排序,如果不是一定是堆排序,然后再交换元素(找交换元素的时候要判断后面已经哪些是有序了从而判断堆元素的个数)执行一次down操作就好了。
#include<iostream>
#include<algorithm>
using namespace std;
int n,cnt;
int a[110],t[110];
bool match(int u){
for(int i = u+1;i <= n;i++){
if(a[i] != t[i]) return false;
}
return true;
}
void down(int u){
int p = u;
if(2*u <= cnt){
if(t[p] < t[2*u])
p = 2*u;
}
if(2*u+1 <= cnt){
if(t[p] < t[2*u+1])
p = 2*u+1;
}
swap(t[u],t[p]);
if(p != u)
down(p);
}
int main(){
cin>>n;
for(int i = 1;i <= n;i++) cin>>a[i];
for(int i = 1;i <= n;i++) cin>>t[i];
int k = 1;
for(k = 1;k + 1 < n;k++){
if(t[k+1] < t[k]) break;
}
if(match(k)){
puts("Insertion Sort");
sort(t+1,t+k+2);
cout<<t[1];
for(int i = 2;i <= n;i++) cout<<" "<<t[i];
}else{
puts("Heap Sort");
cnt = n;
int k = 0;
for(int i = n;i > 0;i--){
if(t[i-1] > t[i]) break;
else k++;
}
cnt -= k;
swap(t[1],t[cnt--]);
down(1);
cout<<t[1];
for(int i = 2;i <= n;i++) cout<<" "<<t[i];
}
return 0;
}