[leetcode]703. Kth Largest Element in a Stream

最新推荐文章于 2021-02-18 21:26:05 发布

你看见我的代码了么

最新推荐文章于 2021-02-18 21:26:05 发布

阅读量1.3k

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode heap 文章标签： leetcode heap

本文链接：https://blog.youkuaiyun.com/weixin_32135877/article/details/81167819

LeetCode 同时被 2 个专栏收录

336 篇文章

订阅专栏

heap

1 篇文章

订阅专栏

本文介绍了一种使用最小堆（priority_queue）的数据结构实现来解决LeetCode上的703题：在一个流中查找第K大的元素。该方法能够高效地维护一个大小为K的最小堆，以快速找到当前流中的第K大元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

[leetcode]703. Kth Largest Element in a Stream

Analysis

年纪大了就会比较惜命~—— [ummmm~]

Design a class to find the kth largest element in a stream. Note that it is the kth largest element in the sorted order, not the kth distinct element.
Your KthLargest class will have a constructor which accepts an integer k and an integer array nums, which contains initial elements from the stream. For each call to the method KthLargest.add, return the element representing the kth largest element in the stream.
这道题是要我们实现一个结构体，给定一个数组，然后每次调用add(val)的时候会向这个数组中添加一个数val，并且输出此时数组中第K大的数。
用最小堆来做，即priority_queue，每次输出最小的那个数就行了。

Implement

class KthLargest {
public:
    KthLargest(int k, vector<int> nums) {
        size = k;
        for(auto num:nums){
            pq.push(num);
            if(pq.size() > size)
                pq.pop();
        }
    }

    int add(int val) {
        pq.push(val);
        if(pq.size() > size)
            pq.pop();
        return pq.top();
    }
private:
    priority_queue<int, vector<int>, greater<int> > pq;
    int size;
};

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