数据流中的中位数

题目描述

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

参照剑指offer，用最大堆和最小堆实现

class Solution {
public:
    void Insert(int num)
    {
        if((max.size()+min.size())%2==0){//总长度是偶数，则新来的数字插入到最小堆
            if(max.size()>0&&num<max[0]){
                max.push_back(num);
                push_heap(max.begin(),max.end(),less<int>());
                num=max[0];
                pop_heap(max.begin(),max.end(),less<int>());
                max.pop_back();
            }
            min.push_back(num);
            push_heap(min.begin(),min.end(),greater<int>());
        }
        else{
            if(min.size()>0&&num>min[0]){
                min.push_back(num);
                push_heap(min.begin(),min.end(),greater<int>());
                num=min[0];
                pop_heap(min.begin(),min.end(),greater<int>());
                min.pop_back();
            }
            max.push_back(num);
            push_heap(max.begin(),max.end(),less<int>());
        }
    }
 
    double GetMedian()
    {
        if(max.size()==0&&min.size()==0)
            return -1;
        if((max.size()+min.size())&1==1)
            return min[0];
        else
            return (max[0]+min[0])/2.0;
    }
    vector<int> max;//左边最大堆
    vector<int> min;//右边最小堆
};