思路
初始化树
将无序数组有序化,通常是升序用大顶堆,降序用小顶堆(反着想
将新的数放在末梢,如果比父结点大,则交换上去
最后把6换上去有:
构造目标顶堆
将堆顶元素和末尾元素交换,那么大数就到了末尾,小的回到了堆顶,再与左右二子对比,与比它大的数继续交换过来,那么大数又回到堆顶,循环上述操作,大数就逐渐放在了末尾处。
下面是从小到大排列的代码:
#include<bits/stdc++.h>
using namespace std;
int main(){
int n,i,k,a[10001],t;
//读入一维数组
cin>>n;
for(i=1;i<=n;i++){
cin>>a[i];
}
//构造大顶堆
for(i=2;i<=n;i++){ //初始有一个a[1]
k=i; //开始逐个添加到树
while(k>0){
if(k/2<=0||a[k]<=a[k/2]) break;
else{
swap(a[k],a[k/2]);
k/=2;
}
}
}
//把大数往后移,构造小顶堆
for(i=n;i>0;i--){
swap(a[1],a[i]);
k=1;
while(k<i){
t=k;
if(2*k<i&&a[2*k]>a[t]) t=2*k;
if(2*k+1<i&&a[2*k+1]>a[t]) t=2*k+1;
if(k==t) break;
swap(a[k],a[t]);
k=t;
}
}
//输出
for(i=1;i<=n;i++){
cout<<a[i]<<" ";
}
return 0;
}
下面是从大到小的代码:
#include<bits/stdc++.h>
using namespace std;
int main(){
int n,i,k,a[10001],t;
//读入一维数组
cin>>n;
for(i=1;i<=n;i++){
cin>>a[i];
}
//构造小顶堆
for(i=2;i<=n;i++){ //初始有一个a[1]
k=i; //开始逐个添加到树
while(k>0){
if(k/2<=0||a[k]>=a[k/2]) break;
else{
swap(a[k],a[k/2]);
k/=2;
}
}
}
//把小数往后移,构造大顶堆
for(i=n;i>0;i--){
swap(a[1],a[i]);
k=1;
while(k<i){
t=k;
if(2*k<i&&a[2*k]<a[t]) t=2*k;
if(2*k+1<i&&a[2*k+1]<a[t]) t=2*k+1;
if(k==t) break;
swap(a[k],a[t]);
k=t;
}
}
//输出
for(i=1;i<=n;i++){
cout<<a[i]<<" ";
}
return 0;
}
示意图及思路引用:
【1】图解排序算法(三)之堆排序
【2】堆排序算法(图解详细流程)