TOP K 问题

Leecode.215
在未排序的数组中找到第k个最大的元素
输入：[3,2,1,5,6,4] 和 k=2
输出：5
输入：[3,2,3,1,2,4,5,5,6] 和 k=4
输出：4

思路：维护一个k大小的最小堆，当堆中元素个数小于k时，新元素直接进入堆中；否则，当堆顶元素小于新元素时，弹出堆顶元素，将新元素加入堆中。遍历完整个数组后，位于堆顶的元素即为第k大的数。

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        priority_queue<int, vector<int>, greater<int>> min_heap; //最小堆
        //最大堆创建  priority_queue<int, vector<int>, less<int>> max_heap;
        for(auto num:nums){
            if(min_heap.size()<k) 
            min_heap.push(num);
            else if(min_heap.top()<num){
                min_heap.pop();
                min_heap.push(num);
            }
         }
         return min_heap.top();
    }
};

Leecode…前 K 个高频元素

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int>mp;        priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>>q;
        for(auto num:nums){
            mp[num]++;
        }
        for(auto begin=mp.begin();begin!=mp.end();begin++){
            if(q.size()<k){
                q.push(make_pair(begin->second,begin->first));
            }
            else{
                if(begin->second>q.top().first){
                    q.pop();
                    q.push(make_pair(begin->second,begin->first));
                }
            }
        }
        vector<int>res;
        while(!q.empty()){
            int tmp=q.top().second;
            q.pop();
            res.push_back(tmp);
        }
        return res;
    }
};