题目链接:https://leetcode-cn.com/problems/top-k-frequent-elements/comments/
题目描述
给定一个非空的整数数组,返回其中出现频率前 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
时间复杂度O(n logk);空间复杂度O(n)
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
unordered_map<int, int> count;
for (auto num:nums)
++count[num];
multimap<int, int, greater<int>> m;
for (auto f:count)
m.insert(make_pair(f.second, f.first));
vector<int> ret;
for(auto it = m.begin();it!=m.end() && k; ++ it,--k)
ret.push_back(it->second);
return ret;
}
};
2 priority_queue
时间复杂度O(nlogk);空间复杂度O(n)
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
using E = std::pair<int, int>;
priority_queue<E, vector<E>, greater<E>> minHeap; // 求topK大用小根堆
map<int, int> count;
for(auto e: nums)
++count[e];
for (auto x:count) {
auto p = make_pair(x.second, x.first);
if(minHeap.size() == k){
if(p < minHeap.top()) continue;
minHeap.pop();
}
minHeap.push(p);
}
vector<int> ret(k);
while (!minHeap.empty()){
ret[--k] = minHeap.top().second;
minHeap.pop();
}
return ret;
}
};