【Leetcode】703. Kth Largest Element in a Stream

原创已于 2023-01-29 04:35:16 修改 · 248 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #数据结构 #c++

于 2020-05-26 07:03:34 首次发布

LC 树、分治与堆专栏收录该内容

270 篇文章

订阅专栏

本文介绍了一种使用最小堆数据结构在线计算数据流中第K大元素的方法。通过维护一个大小为K的最小堆，可以高效地获取到当前数据流中的第K大元素，适用于实时数据处理场景。

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

题目地址：

https://leetcode.com/problems/kth-largest-element-in-a-stream/

要求设计一个数据结构，可以在线计算数据流里第 $k$ 大的数（这里第 $k$ 大的意思是从大到小排好序后位于第 $k$ 位的数，从 $1$ 开始数）。

思路是用最小堆，先将堆里放满 $k$ 个数，这样堆顶即为第 $k$ 大的数。再来新数时，只有比堆顶大，才让其替换堆顶从而进堆；否则略过。这样堆顶就维持住一直等于 $k$ 大的数了。代码如下：

class KthLargest {
 public:
  priority_queue<int, vector<int>, greater<>> heap;
  int k;
  KthLargest(int k, vector<int>& a) {
    this->k = k;
    for (int x : a) {
      heap.push(x);
      if (heap.size() > k) heap.pop();
    }
  }

  int add(int val) {
    heap.push(val);
    if (heap.size() > k) heap.pop();
    return heap.top();
  }
};