《每日一题》——347. 前 K 个高频元素

最新推荐文章于 2024-06-27 16:50:35 发布

shawn&lycheeo

最新推荐文章于 2024-06-27 16:50:35 发布

阅读量54

点赞数

分类专栏：学习之旅文章标签：算法 c++ 数据结构

本文链接：https://blog.csdn.net/qq_40911499/article/details/130244270

版权

学习之旅专栏收录该内容

100 篇文章 0 订阅

订阅专栏

该代码示例展示了如何在C++中利用unordered_map统计数组中每个元素的频率，然后通过小根堆（优先队列）找到出现频率前k大的元素。优先队列默认为大根堆，但通过greater比较器可以构建小根堆，确保堆顶始终是最小的元素。最终将堆中的元素依次出堆，得到频率最高的k个元素。

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

class Solution {
public:
//求前k大，用小根堆（堆顶元素是最小的）
//优先队列默认生成大根堆；priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int,int>>> pq;生成小根堆
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int, int> mp;

        for (auto i : nums){
            mp[i]++;
        }

        priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int,int>>> pq;

        for (auto [num, count] : mp){
            pq.emplace(count, num);
            if (pq.size() > k) pq.pop();
        }

        vector<int> ret;
        while (!pq.empty()) {
            ret.emplace_back(pq.top().second);
            pq.pop();
        }
        return ret;
    }
};