- 基本思想:
1、创建一个堆 H[0……n-1];
2、把堆首(最大值)和堆尾互换;
3、把堆的尺寸缩小 1,并调用 shift_down(0),目的是把新的数组顶端数据调整到相应位置;
4、重复步骤 2,直到堆的尺寸为 1。
代码:
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <iostream>
using namespace std;
int a[100],n;
//不使用递归建立小根堆(效率更高)
void slidedown(int i,int n){
//定义变量用来储存最小值的下标
int t;
//当左节点存在且左节点的值小于父节点时,将左节点的下标赋给t
while(i*2<=n){
if(a[i*2]<a[i]){
t=i*2;
}
//当左节点的值大于父节点时,不做更改
else{
t=i;
}
//如果右节点存在且右节点的值小于父节点(这里的父节点是已经和左节点比过大小后的父节点),
//将右节点的下标赋给t
if(i*2+1<=n && a[i*2+1]<a[t]){
t=i*2+1;
}
//当父节点本身就是三个节点中的最小值时,跳出循环
if(t==i){
break;
}
//当t值有所变化时(也就是原来的父节点并不满足小根堆条件时)
else{
//将父节点替换为三个节点中的最小值
swap(a[t],a[i]);
i=t;
}
}
}
//使用递归建立小根堆 (效率低)
//void slidedown(int i,int n){
//定义变量用来储存最小值的下标
//int t;
//当左节点存在且左节点的值小于父节点时,将左节点的下标赋给t
//if(i*2<=n && a[i*2]<a[i]){
// t=i*2;
//}
//当左节点的值大于父节点时,不做更改
// else{
// t=i;
// }
//如果右节点存在且右节点的值小于父节点(这里的父节点是已经和左节点比过大小后的父节点),
//将右节点的下标赋给t
// if(i*2+1<=n && a[i*2+1]<a[t]){
// t=i*2+1;
// }
//当t值有所变化时(也就是原来的父节点并不满足小根堆条件时)
// if(t!=i){
//将父节点替换为三个节点中的最小值
// swap(a[t],a[i]);
// i=t;
//继续向下行进,形成小根堆
// slidedown(i,n);
// }
//}
//堆排序
void heap_sort(int n){
//构建小根堆
for(int i=n/2;i>=1;i--){
slidedown(i,n);
}
//定义变量存储数组长度
int num=n;
for(int i=1;i<=n;i++){
//打印每次形成小根堆后的根节点(根节点为整个小根堆的最小值)
printf("%d ",a[1]);
//交换根节点的值
swap(a[1],a[num--]);
//继续重复形成小根堆,从而找出第2、3、4……小的值
slidedown(1,num);
}
printf("\n");
return;
}
int main(){
scanf("%d",&n);
srand(unsigned(time(NULL)));
for(int i=1;i<=n;i++){
a[i]=rand()%100+1;
printf("%d ",a[i]);
}
printf("\n");
heap_sort(n);
}