leetcode-347、前K个高频元素

最新推荐文章于 2024-09-20 11:13:13 发布

笨笨小菜

最新推荐文章于 2024-09-20 11:13:13 发布

阅读量71

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/shushi6969/article/details/114362088

版权

leetcode 专栏收录该内容

210 篇文章 0 订阅

订阅专栏

https://leetcode-cn.com/problems/top-k-frequent-elements/

思路：priority_queue和unordered_map使用

struct cmp {
    bool operator() (pair<int, int> a, pair<int, int> b) {
        return a.second < b.second;
    }
};

vector<int> topKFrequent(vector<int>& nums, int k) {
    priority_queue<pair<int, int>, vector<pair<int, int> >, cmp> pri_q;
    unordered_map<int, int> un_mp;
    vector<int> res;
    for (int i = 0;i < nums.size(); i++) {
        un_mp[nums[i]] += 1;
    }
    for (unordered_map<int, int>::iterator it = un_mp.begin() ;it != un_mp.end() ; it++) {
        pri_q.push(make_pair(it->first, it->second));
    }
    int j = 0;
    while (!pri_q.empty() && j < k) {
        j += 1;
        res.push_back(pri_q.top().first);
        pri_q.pop();
    }
    return res;

}