朝花夕拾,重新学 c++ 的库函数了,优先队列:
基本数据类型
定义 priority_queue 只需要写上头文件 #include<queue>
即可
定义,除了最基本的 priority_queue<int>qu
,还可以定义队列的优先级顺序:
priority_queue<int, vector<int>, greater<int>> qu1;// 表示数字小的优先级高
priority_queue<int, vector<int>, less<int>> qu2;// 表示数字大的优先级高
其中,第二个参数,是定义承载底层数据结构堆中的容器
第三个参数是对第一个参数的比较类
如果是 less 说明是大的优先
反之,如果是 greater 说明是小的优先
举例:
此时输出 2
结构体
如果不想设置基本数据类型,而是想设置结构体等,方法如下:
以下代码:Node 是一个结构体,friend 为友元,后面的是重载的比较方法,类似 java
注意:重载只需要用小于号,如果用大于号会报错,f1 > f2 等价于 f2 < f1,f1 等于 f2 等价于 !(f1 < f2) && !(f2 < f1)
#include<iostream>
#include<string>
#include<algorithm>
#include<bits/stdc++.h>
#include<stack>
#include<set>
#include<vector>
#include<map>
#include<queue>
#include<deque>
#include<cctype>
#include<unordered_set>
#include<unordered_map>
#include<fstream>
#include<cstring>
using namespace std;
struct Node{
string name;
int price;
friend bool operator < (Node n1, Node n2) {
// 当价格相同的时候,比较姓名,字典序
if (!(n1.price < n2.price) && !(n2.price < n1.price)) {
return n2.name < n1.name;
}
return n2.price < n1.price;
}
}n1, n2, n3;
int main() {
priority_queue<Node> qu;
n1.name = "12";
n1.price = 12;
n2.name = "11";
n2.price = 12;
n3.name = "34";
n3.price = 34;
qu.push(n1);
qu.push(n2);
qu.push(n3);
while (!qu.empty()) {
cout << qu.top().name << qu.top().price << endl;
qu.pop();
}
}
读者可能会发现,优先队列的排序规则,和 sort 的 cmp 类似,但是有些不一样,其实, 优先队列的这个函数与 sort 的 cmp 函数效果是相反的,如果 cmp 定义的 return a < b 那么就是升序,但是对于 优先队列,就是降序了。
如果结构体内的数据较为庞大,建议使用引用来提高效率
friend bool operator < (const Node &n1, const Node &n2) {
if (!(n1.price < n2.price) && !(n2.price < n1.price)) {
return n2.name < n1.name;
}
return n2.price < n1.price;
}
加上 const 和 & 标志即可