结构体优先队列的定义

前言

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

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

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

正文

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

  1. 大根堆(默认):
priority_queue <int> q;
  1. 小根堆:
priority_queue <int, vector<int>, greater<int> > q;

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

有两个变量

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

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

具体写法是这样的:

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

typedef pair<int, int> P;
  1. 大根堆
priority_queue <P> q;
  1. 小根堆(同上面)
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\) 这种写法,结构体里都是大于号。

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

至于为什么,呵呵。

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

上一篇:POJ 1852 java


下一篇:KNN算法实现对iris数据集的预测