懒,不想说堆是啥了,就说说咋实现一个大顶堆吧
用一个数组保存一个二叉堆,就像二叉树一样,一个节点(下标为
i
n
d
e
x
index
index),其儿子节点分别为
i
n
d
e
x
×
2
和
i
n
d
e
x
×
2
+
1
index\times2和index\times2+1
index×2和index×2+1
如果是一个大顶堆,就满足每个节点的值都比它的两个儿子的值大,显然堆顶就是最大值,对应的就是
h
e
a
p
[
1
]
heap[1]
heap[1]
入堆操作:
将一个元素,放入堆尾部,(即
h
e
a
p
[
+
+
i
n
d
e
x
]
=
v
a
l
heap[++index]=val
heap[++index]=val),然后从堆的底部开始向上(父节点)更新整个堆,因为是大顶堆,因此如果节点的值比父节点大,就交换,直到不比父节点大或者到达堆顶为止。
出堆操作:
出堆肯定是堆顶元素出来,将堆顶元素和最后一个元素交换,然后记录数量的下标
i
n
d
e
x
−
−
index--
index−−就删掉了,删除后要更新堆,这次是自上而下更新,取左右两个儿子的最大值,和当前值比较,如果当前值小,就对应的交换值,注意判断下标是否合法,直到当期的值大于子节点或者到达了堆的底部为止。
/*
heap
大顶堆
*/
#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int heap[N];
int ind;
void up(int n){
while (n > 1){// 堆顶警告
if (heap[n] > heap[n / 2]){// 当前值大与父节点就换
swap(heap[n], heap[n / 2]);
n = n / 2;
}
else break;
}
}
void push(int val){
// 插入采用向上交换
heap[++ind] = val;
int i = ind;
up(ind);
}
void down(int n){
while (n <= ind){// 到达堆底部
int Son = n * 2;// 左儿子
if (Son > ind)break;// 如果没有左儿子就更没有右儿子
if (Son + 1 <= ind && heap[Son] < heap[Son + 1]){
Son++;// 两个儿子都有取一个大的
}
if (heap[n] < heap[Son]){// 当前值小就交换
swap(heap[n], heap[Son]);
n = Son;
}
else break;
}
}
void pop(){
swap(heap[ind], heap[1]);
ind--;
down(1);
}
int top(){
return heap[1];
}
int main()
{
ind = 0;
push(2);
push(5);
push(7);
push(10);
push(8);
push(9);
while (ind){
printf("%d ", top());
pop();
}
return 0;
}