栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > C/C++/C#

STL 优先队列 priority

C/C++/C# 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

STL 优先队列 priority

在优先队列中,元素被赋予优先级。当访问元素时,具有最高优先级的元素最先删除。优先队列是一个队列,具有队列的所有特性,包括队列的基本操作,在这基础上添加了内部的一个排序(堆排序),它本质是一个堆实现的。

priority_queue

当需要用自定义的数据类型时才需要传入 Type, Container, Functional 这三个参数,使用基本数据类型时,只需要传入数据类型,默认是大顶堆。

Type: 数据类型
Container: 容器类型(Container必须是用数组实现的容器,比如vector, deque等等,但不能用 list,STL 里面默认用的是 vector)
Functional: 比较的方式

// 升序队列,小顶堆
priority_queue ,greater > q;
// 降序队列,大顶堆
priority_queue ,less >q;

// 对于基础类型 默认是大顶堆 
// 等同于 priority_queue, less > a;
priority_queue a;

前 K 个高频元素

class Solution {
public:
    vector topKFrequent(vector& nums, int k) {
        vector res;
        unordered_map freq;
        priority_queue, vector>, greater>> pq;
        
        for (auto num : nums) ++freq[num];
        
        for (auto& pair : freq) {
            pq.emplace(pair.second, pair.first);
            if (pq.size() > k) pq.pop();
        }
        
        while (!pq.empty()) {
            res.emplace_back(pq.top().second);
            pq.pop();
        }
        
        return res;
    }
};
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/768126.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号