[leetcode] 295. Find Median from Data Stream

齐豪

于 2017-10-23 11:26:13 发布

阅读量451

点赞数

分类专栏： STL

本文链接：https://blog.youkuaiyun.com/qq_33096883/article/details/78317005

版权

STL 专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一种使用两个堆（最大堆和最小堆）的数据结构来维护数据流中的所有数值，并能高效地找到这些数值的中位数的方法。通过不断调整两个堆的平衡状态，确保能够快速响应中位数的查询请求。

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

https://leetcode.com/problems/find-median-from-data-stream/solution/

class MedianFinder {
    priority_queue<int> lo;                              // max heap
    priority_queue<int, vector<int>, greater<int>> hi;   // min heap

public:
    // Adds a number into the data structure.
    void addNum(int num)
    {
        lo.push(num);                                    // Add to max heap

        hi.push(lo.top());                               // balancing step
        lo.pop();

        if (lo.size() < hi.size()) {                     // maintain size property
            lo.push(hi.top());
            hi.pop();
        }
    }

    // Returns the median of current data stream
    double findMedian()
    {
        return lo.size() > hi.size() ? (double) lo.top() : (lo.top() + hi.top()) * 0.5;
    }
};