给定一个整数数组 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 个高频元素的集合是唯一的
进阶:所设计算法的时间复杂度 必须 优于 O(n log n) ,其中 n 是数组大小。
关联型容器(O(nlogn))
class Solution {
public:vector<int> topKFrequent(vector<int>& nums, int k) {map<int, int> numFre;for (int i = 0; i < nums.size(); i++) {if (numFre.find(nums[i]) == numFre.end()) numFre[nums[i]] = 1;else numFre[nums[i]]++;}multimap<int, int,greater<int>> freNum;for (auto num : numFre) freNum.insert({num.second, num.first});vector<int> res;for (auto num: freNum){res.push_back(num.second);if (!(--k)) break;}return res;}
};