前言
蒟蒻在定义优先队列时,经常忘记重载运算符之后优先队列该定义大根堆还是小根堆。
每次都得试一下 \(QwQ\)。
所以写一篇博客来记录一下,以后就不用试了。
正文
普通的优先队列有两种定义方法。
- 大根堆(默认):
priority_queue <int> q;
- 小根堆:
priority_queue <int, vector<int>, greater<int> > q;
但是这只是一个变量的情况下,当有多个变量怎么办呢?
有两个变量
如果只有两个变量的话,我们可以选择定义pair<first, second>
。
如果是 \(pair\) 类型的优先队列,那么它会按照 .first
比较并排序。
具体写法是这样的:
首先定义一下 \(pair\),方便后面写
typedef pair<int, int> P;
- 大根堆
priority_queue <P> q;
- 小根堆(同上面)
priority_queue <P, vector<P>, greater<P> > q;
有两个或多个变量
我们也可以选择结构体 + 重载运算符来实现。
具体来讲,就是重载运算符告诉程序要判断哪个变量。
这个如何定义小根堆呢?
我们以 \(dijkstra + 堆优化\) 为例:
struct Que{
int x, dis;
bool operator < (const Que &b) const{
return dis > b.dis;//注意这里是大于号
}
// friend bool operator < (Que a, Que b){
// return a.dis > b.dis;
// }
};
priority_queue <Que> q;
注意:不管是不是用 \(friend\) 这种写法,结构体里都是大于号。
大根堆的话自然把大于号改为小于号即可。
至于为什么,呵呵。
直观感觉就是 大于的大于 然后就变成 小于 了吧……