leetcode-295-Find Median from Data Stream

问题

题目：[leetcode-295]

思路

基本思路就是将前半部分元素维护在大根堆当中。
后半部分元素维护在小根堆当中。

但是要注意的是，大根堆和小根堆可以维护各自的有序性。但是，这两个部分的有序性需要在每次插入的时候特别判断以下。
每次，都进行维护。这样就不会出现某一次插入需要弹出很多元素的情形。

代码

class MedianFinder {
public:
    /** initialize your data structure here. */
    MedianFinder() : size(0) {}

    void addNum(int num) {
        if( size & 0x1 ) {
            if( !max_heap.empty() && num < max_heap.top() ) {
                int e = max_heap.top();
                max_heap.pop();

                min_heap.push(e);
                max_heap.push(num);
            }
            else
                min_heap.push(num);
        }
        else{
            if( !min_heap.empty() && num > min_heap.top()  ) {
                int e = min_heap.top();
                min_heap.pop();

                max_heap.push(e);
                min_heap.push(num);
            }
            else
                max_heap.push(num);
        }
        ++size;
    }

    double findMedian() {
        if( size & 0x1 ) return max_heap.top();
        else return ( max_heap.top() + min_heap.top() ) * 1.0 / 2;
    }
private:
    priority_queue< int, vector<int>, less<int> > max_heap;
    priority_queue< int, vector<int>, greater<int> > min_heap;
    int size;
};

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