这个建堆调整过程还是蛮艰辛的。。
#include <stdio.h>
#include <algorithm>
using namespace std;
const int maxn = 100;
//heap为堆,n为元素个数
int heap[11] = {0,5,22,3,45,6,7,88,9,71,16}; //第一个节点存储于数组的1号位,改变这几个字的顺序建成的大顶堆不同
int n = 10;
//对heap数组在[low,high]范围进行向下调整
//其中low为欲调整节点的数组下标,high一般为堆的最后一个元素的数组下标
void downAdjust(int low,int high){
int i = low,j = i*2; //i为欲调整节点,j为左孩子
while(j <= high){ //存在孩子节点
//如果右孩子存在,且右孩子的值大于左孩子
if(j+1 <= high && heap[j+1] >heap[j]){
j = j+1; //让j存储右孩子下标
}
//如果孩子中最大的权值比欲调整节点i大
if(heap[j] >heap[i]){
swap(heap[j],heap[i]); //交换最大权值的孩子与欲调整节点i
i = j; //保持i为欲调整节点,j为i的左孩子
j = i*2; //左孩子
}
else{
break; //孩子的权值均比欲调整节点i小,调整结束
}
}
}
//建堆
void creatHeap(){
for(int i = n/2;i>=1;i--){
downAdjust(i,n);
}
}
//删除堆顶元素
void deleteTop(){
heap[1] = heap[n--]; //用最后一个元素覆盖堆顶元素,并让元素个数减1
downAdjust(1,n); //向下调整堆顶元素
}
//对heap数组在[low,high]范围进行向上调整
//其中low一般设置为1,high表示欲调整节点的数组下标
void upAdjust(int low,int high){
int i = high,j = i/2;
while(j >= low){
if(heap[j] <heap[i]){
swap(heap[j],heap[i]);
i = j;
j = i/2;
}
else{
break;
}
}
}
//添加元素x
void insert(int x){
heap[++n] = x;
upAdjust(1,n);
}
//堆排序
void heapSort(){
creatHeap(); //建堆
for(int i = n;i>1;i--){ //倒着枚举,直到堆中只有一个元素
swap(heap[i],heap[1]); //交换heap[i] 与堆顶
downAdjust(1,i-1); //调整堆顶
}
}
int main(){
printf("初始的元素排列输出:\n");
for(int i = 1;i <=10;i++){ //输出初始的元素排列
printf("%d ",heap[i]);
}
creatHeap();
printf("\n建堆后元素的输出:\n");
for(int i = 1;i <=10;i++){ //输出建堆后的元素排列
printf("%d ",heap[i]);
}
printf("\n堆排序后的元素输出:\n");
heapSort();
for(int i = 1;i <=10;i++){ //输出堆排序后的元素排列
printf("%d ",heap[i]);
}
}
建堆效果如下