347. 前 K 个高频元素

代码无bug

已于 2024-01-18 01:15:51 修改

阅读量316

点赞数 7

分类专栏：优先队列力扣刷题文章标签：算法数据结构 leetcode

于 2024-01-18 01:14:00 首次发布

本文链接：https://blog.csdn.net/aaronyyds/article/details/135663605

版权

力扣刷题同时被 2 个专栏收录

140 篇文章 0 订阅

订阅专栏

优先队列

1 篇文章 0 订阅

订阅专栏

class Solution {
public:
    class compare{
        public:
        bool operator()(pair<int,int>a1,pair<int,int>a2){
            return a1.second>a2.second;
        }
    };
    vector<int> topKFrequent(vector<int>& nums, int k) {
        map<int,int>mp;
        priority_queue<pair<int,int>,vector<pair<int,int> >,compare>q;
        for(auto t:nums){
            mp[t]++;
        }
        for(map<int,int>::iterator it=mp.begin();it!=mp.end();it++){
            q.push(*it);
            if(q.size()>k){
                q.pop();
            }
        }
        vector<int>ans;
        for(int i=0;i<k;i++){
            ans.push_back(q.top().first);
            q.pop();
        }
        return ans;
    }
};