347. 前 K 个高频元素
#队列 2025-01-21 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
class Solution {
private:
struct Compare{
    bool operator()(const pair<int, int>& lhs,const pair<int, int>& rhs){
        return lhs.second < rhs.second;
    }
};
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int, int> map_;
        for (auto num : nums) { // 记录频次
            map_[num]++;
        }

        priority_queue<pair<int, int>, vector<pair<int, int>>,Compare> queue_;
        for (auto it : map_) {  // 加入优先级队列
            queue_.push(it);
        }

        vector<int> result;
        for(int i = 0; i < k; i++){ // 取出靠前的 K 个 记录
            result.push_back(queue_.top().first);
            queue_.pop();
        }

        return result;
    }
};

核心是自定义优先级队列的第三个参数，这用来优先级队列中元素的排序方式。

这边选择实现仿函数，比较两个参数的频率大小，频率越大就越靠前，而频率相等谁前谁后都无所谓了。