【大小堆】剑指 Offer 41. 数据流中的中位数

最新推荐文章于 2024-07-21 11:20:00 发布

Luck66Max

最新推荐文章于 2024-07-21 11:20:00 发布

阅读量186

点赞数

分类专栏：剑指offer II 文章标签： c++ 算法数据结构

本文链接：https://blog.csdn.net/qq_41581765/article/details/129482927

版权

剑指offer II 专栏收录该内容

73 篇文章 0 订阅

订阅专栏

剑指 Offer 41. 数据流中的中位数

在这里插入图片描述

class MedianFinder {
public:
    //数据流，eg：1 2 3 4 5 6，左边是大顶堆，右边是小顶堆
    //当为奇数时，维持右边多一个；偶数时，添加到右边
    priority_queue<int, vector<int>, greater<int>> rightHeap;
    priority_queue<int, vector<int>, less<int>> leftHeap;
    /** initialize your data structure here. */
    MedianFinder() {

    }
    
    void addNum(int num) {
        if(rightHeap.size() == leftHeap.size()){
            leftHeap.push(num);
            int top = leftHeap.top();
            leftHeap.pop();
            rightHeap.push(top);
        }else{
            rightHeap.push(num);
            int top = rightHeap.top();
            rightHeap.pop();
            leftHeap.push(top);
        }
    }
    
    double findMedian() {
        if(rightHeap.size() != leftHeap.size()){
            return rightHeap.top() * 1.0;
        }else{
            return (leftHeap.top() + rightHeap.top()) * 1.0 / 2;
        }
    }
};

/**
 * Your MedianFinder object will be instantiated and called as such:
 * MedianFinder* obj = new MedianFinder();
 * obj->addNum(num);
 * double param_2 = obj->findMedian();
 */