一、简介:
- 头文件:
#include <queue>
- 一般用来解决一些贪心问题,其底层是用堆来实现。
- 最高优先级始终在队首(可以自定义优先级)。
二、优先队列的使用:
1、定义和访问:
定义:
- 定义:
priority_queue<Type, Container, Functional>
- Type 就是数据类型,Container 就是容器类型(Container必须是用数组实现的容器,比如vector,deque等等,但不能用 list。STL里面默认用的是vector),Functional 就是比较的方式,可以自定义。
- 当需要用自定义的数据类型时才需要传入这三个参数 ,使用基本数据类型时,只需要传入数据类型,默认是大顶堆。
访问:
- 和
queue
不一样的是,priority_queue
没有front( )
,back( )
,只能通过top( )
或pop( )
访问队首元素,即优先级最高的元素。
2、常用函数:
- push( ):将
push(x)
是将x加入优先队列,时间复杂度O( log2(n) ),n为当前队列中的元素个数,它会自动调整内部结构。 - top( ):获取队首元素,在使用前先判断队列是否为空。
- pop( ):让队首元素出队,时间复杂度O( log2(n) ),n为当前队列中的元素个数,它会自动调整内部结构。
3、优先队列内元素优先级的设置:
对于基本的数据类型(int、double、char等)来说是根据数字的大小(char是根据字典数的大小)来排序。
- 升序队列
priority_queue<int, vector<int>, greater<int>> q;
- 降序队列
//降序队列
priority_queue<int, vector<int>, less<int>> q;
//等同于
priority_queue<int> q;
4、用pair做优先队列元素的例子:
先比较第一个元素,若相等,则比较第二个元素。
代码:
#include <bits\stdc++.h>
using namespace std;
int main()
{
priority_queue<pair<int, int>> a;
pair<int, int> b(2, 3);
pair<int, int> c(2, 4);
pair<int, int> d(3, 4);
a.push(d);
a.push(c);
a.push(b);
while (!a.empty())
{
cout << a.top().first << ' ' << a.top().second << '\n';
a.pop();
}
}
样例:
3 4
2 4
2 3
5、对结构体排序的自定义规则
注意:他的排序规则与sort种对结构体的排序规则是完全相反的。
第一种:写在结构体内部
struct node
{
int id, s;
bool operator()(const node &a, const node &b)
{
if (a.s != b.s)
return a.s > b.s; //升序
return a.id > b.id; //升序
}
} a;
priority_queue<node, vector<node>, node> q;
第二种:写在结构体外面
struct node
{
int id, s;
} a;
bool operator<(const node &a, const node &b)
{
if (a.s != b.s)
return a.s > b.s; //升序
return a.id > b.id; //升序
}
priority_queue<node, vector<node>, node> q;