leetcode347 前k個高頻元素(Medium)
阿新 • • 發佈:2020-07-18
題目來源:leetcode347 前k個高頻元素
題目描述:
給定一個非空的整數陣列,返回其中出現頻率前 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 是陣列的大小。
題目資料保證答案唯一,換句話說,陣列中前 k 個高頻元素的集合是唯一的。
你可以按任意順序返回答案。
解題思路:
用map儲存數字出現的次數,並進行排序。
class Solution { public: static bool cmp(pair <int,int> p1,pair<int,int> p2){ return p1.second>p2.second; } vector<int> topKFrequent(vector<int>& nums, int k) { map<int,int> m; int i; for(i=0;i<nums.size();i++){ m[nums[i]]++; } vector<int> ans; vector<pair<int,int>> res(m.begin(),m.end()); sort(res.begin(),res.end(),cmp); for(i=0;i<k;i++){ ans.push_back(res[i].first); } return ans; } };
或者也可以用最小堆。