Kth Largest Element in a Stream（C++数据流中的第 K 大元素）

最新推荐文章于 2024-06-03 22:17:31 发布

coolsunxu

最新推荐文章于 2024-06-03 22:17:31 发布

阅读量111

点赞数

分类专栏： C++ LeetCode 文章标签： leetcode c++

本文链接：https://blog.csdn.net/coolsunxu/article/details/109987742

版权

C++ 同时被 2 个专栏收录

735 篇文章 18 订阅

订阅专栏

LeetCode

459 篇文章 1 订阅

订阅专栏

解题思路：

（1）使用优先队列，将其改为小根堆，为后期的第K大元素排在队首

（2）参考网址：https://leetcode.com/problems/kth-largest-element-in-a-stream/discuss/790948/C%2B%2B-Min-Heap-Solution

class KthLargest {
private:
    struct comp {
        bool operator()(const int &a,const int &b) {
            return a>b;
        }
    };
    
    priority_queue<int,vector<int>,comp> pq;
    int size;

public:
    KthLargest(int k, vector<int>& nums) {
        size = k;
        for(auto &w:nums) {
            pq.push(w);
            if(pq.size()>size) pq.pop();
        }
    }
    
    int add(int val) {
        pq.push(val);
        if(pq.size()>size) pq.pop();
        return pq.top();
    }
};

/**
 * Your KthLargest object will be instantiated and called as such:
 * KthLargest* obj = new KthLargest(k, nums);
 * int param_1 = obj->add(val);
 */