优先对列与普通队列不同的是,它能够赋予元素优先级,可以取出最大的数,而普通队列是先进先出,优先队列同时也可以插入数据。
优先队列的实现用到二叉堆的数据结构。
在堆中儿子节点的值不能小于父亲节点的值。排列顺序也是从左到右,从上到下。
在堆中插入数值时,先将数值插入堆的末尾,然后根据规则,找到一个没有大小颠倒的位置。
取出堆中的数据时,先取出根节点的数据,然后用末尾的节点替换根节点,并删除末尾节点,然后与两个儿子节点比较,若两个都小于根节点,选择最小的那个与根节点交换,直到堆中节点都符合规则。
堆操作的时间复杂度
堆(优先队列)的基本操作也是push和pop,由于二叉树的关系,时间复杂度也是O(log N)。
堆的实现
左孩子节点编号=父亲节点编号*2+1
右孩子节点编号=父亲节点编号*2+2
//x为要插入的数值,sn为当前末尾节点编号
void push(int x){
//i为当前查入节点编号
int i=++sn;
while(i>0){
int p=(i-1)/2;
if(heap[p]<=x)
break;
heap[i]=heap[p];
i=p;
}
heap[i]=x;
}
int pop(){
//最小值
int min=heap[0];
int x=heap[sn];
int i=0;
while(i*2+1<sn){
int a=i*2+1;
int b=i*2+2;
if(b<sn&&heap[b]<heap[a])
a=b;
if(heap[a]>=x)
break;
heap[i]=heap[a];
i=a;
}
heap[i]=x;
return min;
}
在C++ STL中已经包含了优先队列priority_queue
要先声明#include <queue>