优先队列又名二叉堆,是特殊的二叉树,二叉堆有两种:最大堆和最小堆
在STL中用priority_queue()表示
最大堆(大根堆):父结点的值总是大于等于任何一个子结点的值
最小堆(小根堆):父结点的值总是小于等于任何一个子结点的值
建堆
以建立小根堆为例
把n个元素建立一个堆,把这n个结点从顶向下,从左到右从1到n编码,目的是把这n个顶点转化为一颗完全二叉树,从最后一个非叶结点(编号为n/2)到第一个结点(编号为1),扫描所有结点,根据需要向下调整。
代码:
#include<bits/stdc++.h>
using namespace std;
int n;
void creat()
{
int i;
for(int i=n/2;i>=1;i--)
siftdown(i);//根据需要向下调整
}
int main()
{
int h[100];
for(int i=0;i<n;i++)
cin>>h[i];//依次编码
creat();
}
插入元素
以下面这个最小堆为例,插入元素24
可以用数组存这个堆,左儿子=父结点乘2,右儿子等于父结点乘2+1,先把插入元素放在数组最后一个位置,但是此时不满足二叉堆的性质,将这个插入元素与它的父结点比较,只要比父结点小就交换,直到它比父结点大,或者成为根结点为止,有一个不断向上调整的过程
如图
向上调整:
//向上调整
void siftup(int i) {
int flag = 0, t;
if (i == 1)
return;//如果是堆顶,就返回,不需要调整了
while (i != 1 && flag == 0) {
if (h[i] < h[i / 2])
swap(i, i / 2);
else
flag = 1;
i = i / 2;
}
return;
}
删除堆顶
把最后一个元素强行拿到堆顶,此时不满足二叉堆的性质,然后在它的两个儿子中找到小的一个与其交换直到它比两个左右儿子都小为止,即为向下调整,找三个结点的最小值当父结点
如图
向下调整:
//向下调整
void siftdown(int i)
{
int t,flag=0,n;
while(i*2<=n&&flag==0)//保证至少有一个左儿子
{
if(h[i]>h[i*2])//如果左儿子比当前父结点小
t=i*2;//用t记录较小值的编号
else
t=i;
if(i*2+1<=n)//判断是否有右儿子
{
if(h[t]>h[i*2+1])//如果右儿子的值更小
{
t=i*2+1;//更新较小点的编号
}
}
if(t!=i)
{
swap(t,i);//如果最小的不是父结点自己,交换结点编号
i=t;//更新i为刚才与它交换的儿子的编号,以方便接下来继续调整
}
else
flag=1;//否则说明父结点比左右儿子都要小了,不需要交换
}
return;
}