这里写目录标题
priority queue 允许用户以任何次序将任何元素推入容器,但取出时一定是从优先权最高的元素开始取。 priority queue 的底层机制为binary max heap(最大二叉堆),即默认为大顶堆。
头文件
#include <queue>
priority_queue和queue不同的就在于我们可以自定义其中数据的优先级, 让优先级高的排在队列前面,优先出队。
定义
priority_queue<Type, Container, Functional>
Type 就是数据类型,Container 就是容器类型(Container必须是用数组实现的容器,比如vector,deque等等,但不能用 list。STL里面默认用的是vector),Functional 就是比较的方式,当需要用自定义的数据类型时才需要传入这三个参数,使用基本数据类型时,只需要传入数据类型,默认是大顶堆。
//对于基础类型 默认是大顶堆,等同于 priority_queue<int, vector<int>, less<int> > a;
priority_queue<int> a;
//升序队列,小顶堆
priority_queue <int,vector<int>,greater<int> > q;
//降序队列,大顶堆
priority_queue <int,vector<int>,less<int> >q;
//greater和less是std实现的两个仿函数(就是使一个类的使用看上去像一个函数。其实现就是类中实现一个operator(),这个类就有了类似函数的行为,就是一个仿函数类了)
成员函数
和队列基本操作相同:
top :访问队头元素
empty: 队列是否为空
size :返回队列内元素个数
push :插入元素到队尾 (并排序)
emplace :原地构造一个元素并插入队列(和push不同)
pop :弹出队头元素 (剩下的元素不会重新排序)
swap :交换内容
自定义比较函数(自定义优先级)
#include <queue>
using namespace std;
//方法1
//小顶堆,升序
struct Status {
int val;
ListNode *ptr;
//必须声明为const函数
bool operator < (const Status &rhs) const {
return val > rhs.val;
}
};
priority_queue <Status> q;
//方法2
//大顶堆,降序
struct compare //重写仿函数
{
//必须声明为public,否则会找不到该仿函数
public:
bool operator() (Status a, Status b)
{
return a.x < b.x;
}
};
priority_queue<Status, vector<Status>, compare> f;