Leetcode 347. 前 K 个高频元素

题目

Leetcode 347. 前 K 个高频元素

代码（首刷自解）

小根堆：
使用函数对象greater来生成小根堆

注意：这里的大于号>规定了优先级，表示优先队列后面的元素都要大于优先队列前面的元素，因为优先队列队首的元素优先级最高，优先队列队尾元素的优先级最低，所以大于号>就规定了优先队列后面的元素都要大于优先队列前面的元素（尾部优先级小于首部优先级），也就是形成一个小根堆，升序排序，每次权值最小的会被弹出来。

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        using my_pair = pair<int, int>;
        auto cmp = [](my_pair& a, my_pair& b) {return a.second < b.second;};
        priority_queue<my_pair, vector<my_pair>, decltype(cmp)> q(cmp);
        unordered_map<int, int> mp;
        for(int i = 0; i < nums.size(); i++) {
            mp[nums[i]]++;
        }
        for(auto& p : mp) {
            q.push(p);
        }
        vector<int> ans;
        for(int i = 0; i < k; i++) {
            ans.push_back(q.top().first);
            q.pop();
        }
        return ans;
    }   
};

代码（10.5 二刷自解）

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int, int> mp;
        for(int& num : nums)
            mp[num]++;
        using pr = pair<int, int>;
        auto cmp = [](auto& a, auto& b) {
            return a.second < b.second;
        };
        priority_queue<pr, vector<pr>, decltype(cmp)> q(cmp);
        for(auto& kv : mp) {
            q.push(kv);
        }
        vector<int> ans;
        for(int i = 0; i < k; i++) {
            ans.emplace_back(q.top().first);
            q.pop();
        }
        return ans;
    }
};