【二叉树】堆排序算法的实现
思路初始化树将无序数组有序化,通常是升序用大顶堆,降序用小顶堆(反着想将新的数放在末梢,如果比父结点大,则交换上去构造目标顶堆将堆顶元素和末尾元素交换,那么大数就到了末尾,小的回到了堆顶,再与左右二子对比,与比它大的数继续交换过来,那么大数又回到堆顶,循环上述操作,大数就逐渐放在了末尾处。#include<bits/stdc++.h>using namespace...
·
思路
初始化树
将无序数组有序化,通常是升序用大顶堆,降序用小顶堆(反着想
将新的数放在末梢,如果比父结点大,则交换上去

最后把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】堆排序算法(图解详细流程)
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐



所有评论(0)