给你一个整数数组 nums 和一个整数 k ,请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:
输入: nums = [1], k = 1
输出: [1]
提示:
1 <= nums.length <= 105
k 的取值范围是 [1, 数组中不相同的元素的个数]
题目数据保证答案唯一,换句话说,数组中前 k 个高频元素的集合是唯一的
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/top-k-frequent-elements
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution { public: vector<int> topKFrequent(vector<int>& nums, int k) { unordered_map<int, int> counts; int max_count = 0; for (const int & num : nums) { max_count = max(max_count, ++counts[num]); } vector<vector<int>> buckets(max_count + 1); for (const auto & p : counts) { buckets[p.second].push_back(p.first); } vector<int> ans; for (int i = max_count; i >= 0 && ans.size() < k; --i) { for (const int & num : buckets[i]) { ans.push_back(num); if (ans.size() == k) { break; } } } return ans; } };
原文:https://www.cnblogs.com/qiaozhoulin/p/14939960.html