题目
给定一个非空的整数数组,返回其中出现频率前 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;
}
};