PAT 1588. 插入还是堆排序
原题链接
中等
作者:
orzorz
,
2020-07-17 17:08:21
,
所有人可见
,
阅读 360
#include<iostream>
using namespace std;
const int N = 110;
int a[N],b[N];
int n;
void down(int u,int cnt){
int t=u;
if(u*2<=cnt&&b[t]<b[u*2]) t=u*2;
if(u*2+1<=cnt&&b[t]<b[u*2+1]) t=u*2+1;
if(u!=t){
swap(b[t],b[u]);
down(t,cnt);
}
}
int main(){
cin>>n;
for(int i=1;i<=n;i++) cin>>a[i];
for(int i=1;i<=n;i++) cin>>b[i];
//1.判断是否是插入排序:
//a)前p个已经排好的数是否已经安装从小到大排序
//b)剩下的序列和原序列相同
int p=2;
while(p<=n&&b[p-1]<=b[p]) p++;
int k=p;//记录下排序好和为排序好的分界线
while(p<=n&&b[p]==a[p]) p++;
// cout<<p<<endl;
if(p==n+1){
puts("Insertion Sort");
//继续一步插入排序的方法:
//从分界线开始往前面移动到合适的位置插入
while(k>=2&&b[k-1]>b[k]){
swap(b[k],b[k-1]);
k--;
}
}else{
puts("Heap Sort");
//继续一步堆排序的方法:
//找到排好序和未排好序的分界点(排好序的区域全部大于堆顶),
//然后进行堆排序的常规操作
p=n;
while(p>=1&&b[p]>=b[1]) p--;
swap(b[1],b[p]);
down(1,p-1);
}
cout<<b[1];
for(int i=2;i<=n;i++) cout<<" "<<b[i];
return 0;
}