给你一个整数数组 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, 数组中不相同的元素的个数]
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/top-k-frequent-elements 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public:
static bool cmp(pair<int, int>& m, pair<int, int>& n) {
return m.second > n.second;
}
vector<int> topKFrequent(vector<int>& nums, int k) {
map<int,int> mp;
//统计每个元素出现的频率
for(int i = 0; i < nums.size(); ++i){
mp[nums[i]] ++;
}
// for(auto iter = map.begin(); iter != map.end(); ++iter) {
//cout << iter->first << " " << iter->second << endl;
// }
//由于sort对map不能排序,转换为vector的pair排序,单调递减
vector<pair<int, int> > mp_sort(mp.begin(),mp.end());
sort(mp_sort.begin(),mp_sort.end(),cmp);
//找出前k个出现频率大的
vector<int> res;
auto iter = mp_sort.begin();
while(k--){
res.push_back(iter->first);
iter++;
}
return res;
}
};
|