个人学习笔记,如果帮助到你点个赞再走吧。
首先我们需要了解priority_queue在库<queue> 中,其底层实现采用的是堆排序,基本上是很快的一种排序,在大数据量时可以考虑使用这个来辅助排序,废话少说
和队列基本操作相同:
- top 访问队头元素
- empty 队列是否为空
- size 返回队列内元素个数
- push 插入元素到队尾 (并排序)
- emplace 原地构造一个元素并插入队列
- pop 弹出队头元素
- swap 交换内容
定义:priority_queue<Type, Container, Functional>
Type 就是数据类型,Container 就是容器类型(Container必须是用数组实现的容器,比如vector,deque等等,但不能用 list。STL里面默认用的是vector),Functional 就是比较的方式,当需要用自定义的数据类型时才需要传入这三个参数,使用基本数据类型时,只需要传入数据类型,默认是大顶堆,即队首元素最大。
废话少说上代码,先看主函数,尤其是注释;
#include <iostream>
#include <queue>
#include <vector>
using namespace std;
struct Node
{
int id;
int cost;
};
struct cmp
{
bool operator()(const Node& a,const Node& b)
{
return a.cost > b.cost;
}
};
struct cmp2
{
bool operator()(const Node& a, const Node& b)
{
return a.cost < b.cost;
}
};
int main()
{
priority_queue<int> q;//(什么参数都不填)默认直接创建就是堆顶/队首为最大的元素
//其等价的创建方式
priority_queue<int, vector<int>, less<int>> q1;//还是大顶堆
//记不住就这样,greater是小,都是反着来的,less是大,下面结构体定义函数也是这样
priority_queue<int, vector<int>, greater<int>>q2;//小顶堆,堆顶/队首是最小元素
//来看结构体的
//首先在上面定义了结构体和比较函数,比较函数也是个结构体,其实还是有其他方法
//但是这个最简单,对于考试的同学手撕比较友好
//定义小顶堆,小顶堆是不是对应greater,看看比较结构体cmp是不是也是求大的,一样类似,都是反着记忆
priority_queue<Node, vector<Node>, cmp> q3;
//若要定义结构体大顶堆直接把cmp的符号改为<就行了,此时对应less
priority_queue<Node, vector<Node>, cmp2> q4;
//下面测试一下;
q.push(1);
q.push(2);
q1.push(1);
q1.push(2);
Node a;
Node b;
a.cost = 1;
b.cost = 2;
q3.push(a);
q4.push(a);
q3.push(b);
q4.push(b);
cout << "q1.top():" << q1.top() << endl;
cout << "q2.top():" << q2.top() << endl;
cout << "q3.top():" << q3.top().cost << endl;
cout << "q4.top():" << q4.top().cost << endl;
return 0;
}
总结:自定义时记住反着来就行,大顶堆用less,小的用greater,比较函数cmp也是如此。
至于为什么呢,因为大顶堆每次都出栈顶的话,并查到对应二叉树的后面,最后得到的二叉树是从小到 到大的,详情见堆排序算法。
运行结果如下
无bug,完结撒花,个人学习笔记,帮助到你请点个赞吧。