leetcode 前K个高频元素 （C++）

题目

给定一个非空的整数数组，返回其中出现频率前 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 是数组的大小。

解题思路：

1.借助unordered_map然后统计nums中出现的数字以及对应的频数
2.然后用一个极大堆来保存最大的k个频数
3.最后再遍历一次这个unordered_map，若是频数在最大的k个之中，则将其push到最后的答案向量之中即可。
具体代码如下:

class Solution 
{
public:
    vector<int> topKFrequent(vector<int>& nums, int k) 
    {
        vector<int> res{};
        vector<int> count_set{};
        // vector<int> tmp{};
        unordered_map<int, int> mymap{};
        for (int x : nums)
        {
            if (mymap.find(x) != mymap.end())
            {
                mymap[x] += 1;
            }
            else
            {
                mymap[x] = 1;
            }
        }
        
        for (auto x : mymap)
        {
            count_set.push_back(x.second);
            push_heap(count_set.begin(), count_set.end(), greater<int>());
            if (count_set.size() > k)
            {
                pop_heap(count_set.begin(), count_set.end(), greater<int>());
                count_set.pop_back();
            }
        }
        
        for (auto x : mymap)
        {
            if (find(count_set.begin(), count_set.end(), x.second) != count_set.end() && find(res.begin(), res.end(), x.first) == res.end())
            {
                res.push_back(x.first);
            }
        }
        return res;
    }
};