C++ 优先队列 priority_queue

priority queue 允许用户以任何次序将任何元素推入容器,但取出时一定是从优先权最高的元素开始取。 priority queue 的底层机制为binary max heap(最大二叉堆),即默认为大顶堆。

头文件

#include 

priority_queue和queue不同的就在于我们可以自定义其中数据的优先级, 让优先级高的排在队列前面,优先出队。

定义

priority_queue<Type, Container, Functional>

Type 就是数据类型,Container 就是容器类型(Container必须是用数组实现的容器,比如vector,deque等等,但不能用 list。STL里面默认用的是vector),Functional 就是比较的方式,当需要用自定义的数据类型时才需要传入这三个参数,使用基本数据类型时,只需要传入数据类型,默认是大顶堆。

//对于基础类型 默认是大顶堆,等同于 priority_queue, less > 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 
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,否则会找不到该仿函数
publicbool operator() (Status a, Status b) 
    {
        return a.x < b.x; 
    }
};
priority_queue<Status, vector<Status>, compare> f;

你可能感兴趣的:(c/c++,c++,开发语言)