LeetCode347 前k个高频元素

给定一个非空的整数数组，返回其中出现频率前 k 高的元素。

 

用最小堆解决。先用map记录每个元素的出现次数，然后循环往最小堆中压入，如果堆中节点数量大于k了就弹出堆顶。

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> map;
        for(int i : nums) map[i] ++; //遍历
        priority_queue< pair<int,int>, vector< pair<int,int> >, greater< pair<int,int> > > q; //最小堆
        for(auto it : map) 
            if(q.size() == k) { //队列满了
                if(it.second > q.top().first) { //堆排
                    q.pop();
                    q.push(make_pair(it.second, it.first));
                }
            }
            else q.push(make_pair(it.second, it.first));
        vector<int> res;
        while(q.size()) { //将优先队列中k个高频元素存入vector
            res.push_back(q.top().second);
            q.pop();
        }
        return vector<int>(res.rbegin(), res.rend());
    }
};

作者：OrangeMan
链接：https://leetcode-cn.com/problems/top-k-frequent-elements/solution/cjian-ji-dai-ma-by-orangeman-5/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

这里压入堆中的时候需要反过来压入<second,first>，应该是因为优先队列会根据里面的pair的first进行排列。