295. 数据流的中位数

Posted 2021-09-11 Harris-H

295. 数据流的中位数

支持插入、查询中位数。

构造两个优先队列。

大顶堆存小于等于中位数的数。

小顶堆存大于中位数的数。

添加就看若大顶堆为空或者$x$小于等于大顶堆的堆顶数，就加入到大顶堆，然后维护大顶堆的$size$减去小顶堆$size$之差小于等于1.

#define sz(a) (int)a.size()
class MedianFinder {
public:
    /** initialize your data structure here. */
    priority_queue<int>q1;
    priority_queue<int,vector<int> ,greater<int> >q2;
    MedianFinder() {}
    void addNum(int x) {
        if(q1.empty()||x<=q1.top()){
            q1.push(x);
            if(sz(q2)+1<sz(q1)){
                q2.push(q1.top());
                q1.pop();
            }
        }
        else {
            q2.push(x);
            if(sz(q2)>sz(q1)){
                q1.push(q2.top());
                q2.pop();
            }
        }
    }
    
    double findMedian() {
        if(sz(q1)>sz(q2)) return q1.top();
        else return (q1.top()+q2.top())/2.0;
    }
};

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