#include<bits/stdc++.h>
using namespace std;
const int maxn = 100;
int heap[maxn];//数组存的数据从下标为1开始
int n = 100;
//low为要数组中要调节元素对应的下标,high为数组中最后一个元素所对映的下标
void downAdjust(int low,int high){
//i为父节点,j为左儿子节点
int i = low,j = low*2;
while(i<=high){
//取儿子中更大的那个
if(j+1<=high&&heap[j+1]>heap[j])
j++;
//如果儿子大与父节点,就将其交换,并继续迭代,看节点是否能继续下沉(是否还有儿子节点比其权值更小)
if(heap[j] > heap[i]){
swap(heap[j],heap[i]);
i = j;
j = i * 2;
}//儿子节点都比父节点小,下调的位置找到,跳出循环
else
break;
}
}
void createheap(){
//从第最后一个非叶子节点开始遍历
for(int i=n/2;i>=1;i--){
downAdjust(i,n);
}
}
void heapsort(){
//先将原数组建堆
createheap();
//每次取堆顶元素(交换heap[1],heap[i],将堆顶元素放入最后一个位置),在对第一个位置进行下调
for(int i=n;i>1;i++){//直到堆中就一个元素为止
swap(heap[i],heap[1]);
downAdjust(1,i-1);
}
}
//删除堆顶元素
void deleteTop(){
heap[1] = heap[n--];
downAdjust(1,n);
}
//堆上调(用于插入操作),就是将要调整的节点不断与父节点比较大小,若比父节点大就与其交换,直到到达合适的位置
void upAdjust(int low,int high){
int i = high, j = high / 2;
while(j>=low){
//若父节点小于子节点,就对其进行交换吗,且继续迭代,直到寻得合适的位置
if(heap[j] < heap[i]){
swap(heap[j],heap[i]);
i = j;
j = i / 2;
}
else
break;
}
}
//堆中添加一个元素
void insert(int x){
heap[++n] = x;
upAdjust(1,n);
}
堆的模板
最新推荐文章于 2023-08-02 17:24:10 发布