包含在头文件<queue>里。
声明形式
priority_queue< type, container, function >
这三个参数,后面两个可以省略,第一个不可以。默认是vector容器,以 operator< 为比较方式,默认是大顶堆。
- type:数据类型;
- container:实现优先队列的底层容器;
- function:元素之间的比较方式;
成员函数
bool empty() const
//返回值为true,说明队列为空;
int size() const
//返回优先队列中元素的数量;
void pop()
//删除队列顶部的元素,也即根节点
int top()
//返回队列中的顶部元素,但不删除该元素;
void push(int arg)
//将元素arg插入到队列之中;
- 构造大顶堆
//构造一个空的优先队列(此优先队列默认为大顶堆)
priority_queue<int> big_heap;
//另一种构建大顶堆的方法
priority_queue<int,vector<int>,less<int> > big_heap2;
- 构造小顶堆
//构造一个空的优先队列,此优先队列是一个小顶堆
priority_queue<int,vector<int>,greater<int> > small_heap;
需要注意的是,如果使用less<int>和greater<int>,
需要头文件:#include <functional>
- 自定义比较函数
// 小根堆的回调函数
struct cmp{
bool operator()(ListNode *a,ListNode *b){
return a->val > b->val;
};
priority_queue<ListNode*, vector<ListNode*>, cmp> pri_queue;