0
点赞
收藏
分享

微信扫一扫

347. 前 K 个高频元素(c++)

生态人 2022-04-23 阅读 45
c++

给你一个整数数组 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, 数组中不相同的元素的个数]

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;
    }
};
举报

相关推荐

0 条评论