priority_queue 优先队列

最新推荐文章于 2024-01-28 23:08:01 发布

lucky tiger

最新推荐文章于 2024-01-28 23:08:01 发布

阅读量211

点赞数

分类专栏： C++ leetcode力扣数据结构

本文链接：https://blog.csdn.net/weixin_42269817/article/details/105667917

版权

C++ 同时被 3 个专栏收录

78 篇文章 1 订阅

订阅专栏

leetcode力扣

67 篇文章 0 订阅

订阅专栏

数据结构

24 篇文章 2 订阅

订阅专栏

本文深入探讨了优先队列的实现原理，包括大顶堆和小顶堆的使用，以及相关函数如empty(), size(), pop(), top() 和 push() 的功能。通过具体实例，如力扣面试题17.14最小k个数，展示了如何利用优先队列解决实际问题。

摘要由CSDN通过智能技术生成

1、大顶堆升序排列从小到大排序底层实现为最大堆

priority_queue<int> big_heap;

priority_queue<int,vector<int>,less<int> > big_heap2;

2、小顶堆降序排列、

priority_queue<int,vector<int>,greater<int> > small_heap;

3、相关函数

    bool empty() const
    返回值为true，说明队列为空；
    int size() const
    返回优先队列中元素的数量；
    void pop()
    删除队列顶部的元素，也即根节点
    int top()
    返回队列中的顶部元素，但不删除该元素；
    void push(int arg)
    将元素arg插入到队列之中；

和队列基本操作相同:

top 访问队头元素
empty 队列是否为空
size 返回队列内元素个数
push 插入元素到队尾 (并排序)
emplace 原地构造一个元素并插入队列
pop 弹出队头元素
swap 交换内容

4、模板参数

优先队列有三个参数，其声明形式为：

priority_queue< type, container, function >

这三个参数，后面两个可以省略，第一个不可以。
其中： type：数据类型； container：实现优先队列的底层容器； function：元素之间的比较方式；

对于container，要求必须是数组形式实现的容器，例如vector、deque，而不能使list。
在STL中，默认情况下（不加后面两个参数）是以vector为容器，以 operator< 为比较方式，所以在只使用第一个参数时，优先队列默认是一个最大堆，每次输出的堆顶元素是此时堆中的最大元素。

5、力扣面试题17.14 最小k个数

class Solution {
public:
    vector<int> smallestK(vector<int>& arr, int k) {
        priority_queue<int> que;
        vector<int> res;
        for(auto& a:arr) 
        {
            que.push(a);
            if(que.size()>k)
            {
                que.pop();
            }
        }
    //res.assign(que.begin(),que.end());
    while(!que.empty())
    {
        res.push_back(que.top());
        que.pop();
    }
    return res;

    }
};