LeetCode:703. 数据流中的第K大元素

最新推荐文章于 2022-09-03 09:57:30 发布

原创最新推荐文章于 2022-09-03 09:57:30 发布 · 176 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

273 篇文章

订阅专栏

本文介绍了一种使用最小堆数据结构实现快速查找数组中第K大元素的方法。通过在构造函数中初始化堆，并在add函数中维护堆大小，确保堆顶始终为第K大元素。

本题的解题过程如下，首先要明确题目的意思：本题目要求给定vector在放入最小堆之后第K大的元素，我们可以在添加元素的时候（也就是add函数）检查队中的元素格式是否大于K，一旦发现大于K那么就将多余的数字弹出，当我们将目标值val加入堆中之后，仍然要维护堆的大小，其过程同上，最后将堆顶元素返回即可。

class KthLargest {
public:
    priority_queue<int, vector<int>, greater<int> > q;//定义一个最小堆
    int K;
    KthLargest(int k, vector<int>& nums) {
        //不断地push元素
        K=k;
        for(int i=0;i<nums.size();i++)
        {
            q.push(nums[i]);
            if(q.size()>K)
            {
                q.pop();
            }
        }
    }
    
    int add(int val) {
        while(q.size()>K)
        {
            q.pop();
        }
        q.push(val);
        if(q.size()>K)
        {
            q.pop();
        }
        return q.top();
    }
};

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