结构体优先队列的定义

前言

蒟蒻在定义优先队列时，经常忘记重载运算符之后优先队列该定义大根堆还是小根堆。

每次都得试一下 \(QwQ\)。

所以写一篇博客来记录一下，以后就不用试了。

正文

普通的优先队列有两种定义方法。

1. 大根堆（默认）：

priority_queue <int> q;

2. 小根堆：

priority_queue <int, vector<int>, greater<int> > q;

但是这只是一个变量的情况下，当有多个变量怎么办呢？

有两个变量

如果只有两个变量的话，我们可以选择定义pair<first, second>。

如果是 \(pair\) 类型的优先队列，那么它会按照 .first 比较并排序。

具体写法是这样的：

首先定义一下 \(pair\)，方便后面写

typedef pair<int, int> P;

1. 大根堆

priority_queue <P> q;

2. 小根堆（同上面）

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\) 这种写法，结构体里都是大于号。

大根堆的话自然把大于号改为小于号即可。

至于为什么，呵呵。

直观感觉就是 大于的大于 然后就变成 小于 了吧……