Leecode347 前k个高频元素

最新推荐文章于 2024-09-11 22:48:33 发布

碳基生物

最新推荐文章于 2024-09-11 22:48:33 发布

阅读量93

点赞数

文章标签：算法 leetcode 数据结构

本文链接：https://blog.csdn.net/hhh_huoluobo/article/details/132690695

版权

class Solution {
public:
    static bool compare(pair<int, int>& m, pair<int, int>& n){
        return m.second > n.second;
    }
    vector<int> topKFrequent(vector<int>& nums, int k) {
        int n = nums.size();
        unordered_map<int, int> record;
        for(auto x : nums){
            record[x]++;
        }
        priority_queue<pair<int, int>, vector<pair<int, int>>, decltype(&compare)> pq(compare);
        for(auto x : record){
            pq.emplace(x);
            if(pq.size() > k){
                pq.pop();
            }
        }
        vector<int> res;
        while(!pq.empty()){
            res.push_back(pq.top().first);
            pq.pop();
        }
        return res;
    }
};