C++优先队列(priority_queue)
Posted Billy Miracle
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了C++优先队列(priority_queue)相关的知识,希望对你有一定的参考价值。
首先要包含头文件#include<queue>
, 他和queue
不同的就在于我们可以自定义其中数据的优先级, 让优先级高的排在队列前面,优先出队。优先队列具有队列的所有特性,包括队列的基本操作,只是在这基础上添加了内部的一个排序,它本质是一个堆实现的。
基本操作
它的基本操作和队列基本操作相同:
- top 访问队头元素
- empty 队列是否为空
- size 返回队列内元素个数
- push 插入元素到队尾 (并排序)
- emplace 原地构造一个元素并插入队列
- pop 弹出队头元素
- swap 交换内容
基本使用
定义:priority_queue<Type, Container, Functional>
Type
就是数据类型,Container
就是容器类型(Container
必须是用数组实现的容器,比如vector
,deque
等等,但不能用 list
。STL里面默认用的是vector
),Functional
就是比较的方式。当需要用自定义的数据类型时才需要传入这三个参数,使用基本数据类型时,只需要传入数据类型,默认是大顶堆。
示例:
//升序队列,小顶堆
priority_queue <int,vector<int>,greater<int> > q;
//降序队列,大顶堆
priority_queue <int,vector<int>,less<int> >q;
//greater和less是std实现的两个仿函数(就是使一个类的使用看上去像一个函数。其实现就是类中实现一个operator(),这个类就有了类似函数的行为,就是一个仿函数类了)
例题:
菜鸡思考:
这个题挺离谱的,但不难想到最简单的方法应该是维护两个数组,前一半,后一半。这样应该会比整体排序/插入
再寻找简单一点。但这么一想,其实只时刻需要知道前一个数组的最后一个以及后一个数组的最开始一个就可以了,这个时候就可以使用优先队列啦。
菜鸡标题:
class MedianFinder
private:
priority_queue<int> frontQueue;
priority_queue<int, vector<int>, greater<int>> backQueue;
public:
/** initialize your data structure here. */
MedianFinder()
void addNum(int num)
if (frontQueue.empty())
frontQueue.push(num);
return;
if (backQueue.size() + 1 == frontQueue.size())
if ((backQueue.empty() && frontQueue.top() > num) || num <= frontQueue.top())
backQueue.push(frontQueue.top());
frontQueue.pop();
frontQueue.push(num);
return;
if (num >= frontQueue.top())
backQueue.push(num);
return;
else
if (num <= backQueue.top())
frontQueue.push(num);
else if (num > frontQueue.top())
frontQueue.push(backQueue.top());
backQueue.pop();
backQueue.push(num);
return;
double findMedian()
if (frontQueue.empty())
return 0;
if (backQueue.empty() || frontQueue.size() == backQueue.size() + 1)
return frontQueue.top();
if (frontQueue.size() == backQueue.size())
return ((double)frontQueue.top() + (double)backQueue.top()) / 2;
return 0;
;
/**
* Your MedianFinder object will be instantiated and called as such:
* MedianFinder* obj = new MedianFinder();
* obj->addNum(num);
* double param_2 = obj->findMedian();
*/
以上是关于C++优先队列(priority_queue)的主要内容,如果未能解决你的问题,请参考以下文章
C++ 优先队列 堆 priority_queue的使用 以及内部使用结构化pair的排序的用法 auto在其中的用法(结构化绑定 C++17以上)
C++ 初阶优先级队列(Priority_Queue)底层框架模拟实现