剑指offer——63/66数据流中的中位数

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

思路：数据分为两部分，大顶堆得到最大的数放在min中，小顶堆得到最小的数放在max中。

class Solution {
public:
    vector<int>min;
    vector<int>max;
    void Insert(int num)
    {
        int size=min.size()+max.size();
        if((size&1)==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()
    { 
        int size=min.size()+max.size();
        if(size<=0)
            return 0;
        if((size&1)==0)
            return (max[0]+min[0])/2.0;
        else
            return min[0];
    }

};