剑指offer:数据流中的中位数

题目描述

如何得到一个数据流中的中位数？如果从数据流中读出奇数个数值，那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值，那么中位数就是所有数值排序之后中间两个数的平均值。

class Solution {
public:
    Solution() :cnt(0){}

    void Insert(int num)
    {
        cnt++;

        if (cnt == 1){
            maxHeap.push(num);
        }
        else{
            //这个判断在前，因为minHeap.top()第二次为空
            if (num <= maxHeap.top()){
                maxHeap.push(num);
            }
            else{
                minHeap.push(num);
            }
        }

        if (maxHeap.size() > minHeap.size() + 1){
            int top = maxHeap.top();
            maxHeap.pop();
            minHeap.push(top);
        }

        if (maxHeap.size() < minHeap.size()){
            int top = minHeap.top();
            minHeap.pop();
            maxHeap.push(top);
        }

    }

    double GetMedian()
    {
        double median;
        if (cnt & 1){
            median =static_cast<double>(maxHeap.top());
        }
        else{
            median = (maxHeap.top() + minHeap.top())*1.0 / 2; 
        }

        return median;
    }

private:
    priority_queue<int> maxHeap;
    priority_queue<int, vector<int>, greater<int>> minHeap;
    int cnt;
};