347. 前 K 个高频元素

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

示例 1:

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]

示例 2:

输入: nums = [1], k = 1
输出: [1]

提示：

• 你可以假设给定的 k 总是合理的，且 1 ≤ k ≤ 数组中不相同的元素的个数。
• 你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。
• 题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的。
• 你可以按任意顺序返回答案。

解答

用map记录元素出现的次数，再依次将出现次数最多的元素查找出来

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        map<int, int> m;
        vector<int> result;
        vector<int> order;
        for(int i : nums){
            if(m.count(i))
                m[i]++;
            else
                m[i] = 1;
        }
        for(map<int, int>::iterator iter=m.begin();iter!=m.end();iter++){
            order.push_back(iter->second);
        }
        sort(order.rbegin(), order.rend());
        for(int i=0;i<k;i++){
            for(map<int, int>::iterator iter=m.begin();iter!=m.end();iter++){
                if(iter->second == order[i]){
                    result.push_back(iter->first);
                    iter->second = 0;
                }
            }
        }
        return result;
    }
};

使用优先队列

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        vector<int> result;
        unordered_map<int, int> m;
        priority_queue<pair<int, int>> q;
        for(int i : nums){
            m[i]++;
        }
        for(auto p : m){
            // 默认降序，因此将出现次数反转
            q.push(pair<int, int>(-p.second, p.first));
            if(q.size()>k)
                q.pop();
        }
        while(q.size()){
            result.push_back(q.top().second);
            q.pop();
        }
        return result;
    }
};