347. 前 K 个高频元素(STL)
- 2020 年 2 月 17 日
- 筆記
虽然tag是中等难度,感觉还是比较简单的,不难想到对每一个数记录它的出现次数,然后按照出现次数从大到小排序输出就好了,主要是实现方法,这里我用了的是map+vector+pair,用时已经是我最快的了,那些更快的就看不太懂他们的代码了…

AC代码:
class Solution { public: vector<int> topKFrequent(vector<int>& nums, int k) { vector<int> ans; vector<pair<int,int>> v; unordered_map<int,int> ma; for(int i=0;i<nums.size();i++){ ma[nums[i]] ++; } for(auto i : ma){ v.push_back(make_pair(i.first, i.second)); } sort(v.begin(), v.end(), [](const pair<int,int> a,const pair<int,int> b){ if(a.second == b.second)return a.first < b.first; return a.second > b.second; }); for(int i=0;i<k;i++){ ans.push_back(v[i].first); } return ans; } };