【剑斩OFFER】算法的暴力美学——LeetCode 703 题：数据流中的第 K 大元素

一、题目描述

二、算法原理

思路：维护 k 个节点的小根堆

例如：

三、代码实现

class KthLargest {
public:
    KthLargest(int k, vector<int>& nums) {
        _k = k;
        for(auto& e : nums)
        {
            minI.push(e);
            if(minI.size() > k) minI.pop();//维护 k 个节点的堆
        }
    }
    
    int add(int val) {
        minI.push(val);
        if(minI.size() > _k) minI.pop();
        if(minI.size() < _k) return 0;//没有第 k 大的值
        return minI.top();
    }
private:
    int _k;
    priority_queue<int,vector<int>,greater<int>> minI;
};