給定一個整數數組 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;}
};