题干:给你一个整数数组 nums
和一个整数 k
,请你返回其中出现频率前 k
高的元素。你可以按 任意顺序 返回答案。
思路:
用哈希表(unordered_map)存储数组里的元素和他们出现的次数,然后赋值给pair类型的数组,最后根据pair里的second倒叙排序。
要注意的点:
1、在leetcode里写cmp函数的时候必须要加static
2、cmp函数里面的元素必须是const
3、pair类型的数组的创建方法和遍历方法
具体见《容易遗忘的知识》
代码:
class Solution {
public:
static bool mycmp(const pair<int,int>&p1,const pair<int,int>&p2){
return p1.second>p2.second;
}
vector<int> topKFrequent(vector<int>& nums, int k) {
unordered_map<int,int>map1;
for(int i=0;i<nums.size();i++){
map1[nums[i]]++;
}
vector<int>ar(k);
pair<int,int>store[nums.size()];int i=0;
for(auto it=map1.begin();it!=map1.end();it++){
store[i].first=it->first;
store[i].second=it->second;
i++;
}
sort(store,store+nums.size()-1,mycmp);
for(i=0;i<k;i++){
ar[i]=store[i].first;
}
return ar;
}
};