Leetcode295. 数据流的中位数 -hot100

最新推荐文章于 2024-10-02 22:45:05 发布

meeiuliuus

最新推荐文章于 2024-10-02 22:45:05 发布

阅读量412

点赞数 10

分类专栏： # leetcode --- hard 文章标签：算法 leetcode c++

本文链接：https://blog.csdn.net/qq_52313711/article/details/136590025

版权

leetcode --- hard 专栏收录该内容

21 篇文章 0 订阅

订阅专栏

这篇文章介绍了使用C++实现的MedianFinder类，它利用优先队列(maxHeap和minHeap)来高效计算和维护输入整数列表的中位数。当添加新数时，会动态调整两个堆以保持大小关系，最终在O(logn)时间内找到中位数。

摘要由CSDN通过智能技术生成

题目：

代码(首刷看解析 2024年3月9日）：

class MedianFinder {
private:
    priority_queue<int> maxHeap;
    priority_queue<int, vector<int>, greater<int>> minHeap;
public:
    MedianFinder() {

    }
    
    void addNum(int num) { 
        if (maxHeap.empty() || num < maxHeap.top()) {
            maxHeap.push(num);
        } else {
            minHeap.push(num);
        }

        if (maxHeap.size() > minHeap.size() + 1) {
            minHeap.push(maxHeap.top());
            maxHeap.pop();
        } else if (minHeap.size() > maxHeap.size()) {
            maxHeap.push(minHeap.top());
            minHeap.pop();
        }
    }
    
    double findMedian() {
        if (maxHeap.size() == minHeap.size()) {
            return double(maxHeap.top() + minHeap.top()) / 2.0;
        } else {
            return double(maxHeap.top());
        }
    }
};