C++ priority_queue (优先队列)

简介

---

• 优先队列底层采用“堆”进行实现。
• 在优先队列中，任何时刻队首元素一定是当前队列中优先级最高的元素。
• 优先级最高的元素可能是最大的（大根堆），也可能是最小的（小根堆）。
• 可以不断往优先队列中添加某个优先级的元素，也可以不断弹出优先级最高的那个元素，每次操作其会自动调整结构， 始终保证队首元素的优先级最高。
• 优先队列可以用来解决一些贪心问题。

常用用法

头文件 #include <queue>
需要使用 std 命名空间 using namespace std;

说明	方法	备注
定义优先队列	priority_queue<type> q;	type可以是任何基本类型或者容器
入队	q.push(x)	将元素插入优先队列
出队	q.pop()	移除队首元素
返回队首元素	q.top()	优先级最高元素
判断队列是否为空	q.empty();	为空返回 1
获取队列大小	q.size()	返回队列中元素个数

大顶堆与小顶堆的定义：

//大顶堆（降序，从大到小）
priority_queue <int,vector<int>,less<int> > q;
priority_queue <int> q; //默认大顶堆
//小顶堆（升序，从小到大）
priority_queue <int,vector<int>,greater<int> >q;

结构体优先队列：

struct st
{
	int a; 
	int b;
};
bool operator < (const st &s1,const st &s2)
{
	return s1.a > s2.a; //按a排序，从小到大
}

int main()
{
	priority_queue <st> q; //定义优先队列
	int x=1,y=2;
	q.push({x,y}); //C++11可以这样表示 （a=x b=y）
}

pair 的使用：

先比较 pair 的第一个元素，再比较第二个元素。

#include <iostream>
#include <queue>
using namespace std;
int main() 
{
    priority_queue<pair<int, int> > q;
    pair<int, int> a(1, 2);
    pair<int, int> b(1, 3);
    pair<int, int> c(2, 5);
    q.push(a);
    q.push(b);
    q.push(c);
    while (!q.empty()) 
    {
        cout << q.top().first << ' ' << q.top().second << '\n';
        q.pop();
    }
}

运行结果：

2 5
1 3
1 2