[ATO] 41、数据流中的中位数

题目描述

如何得到一个数据流中的中位数？如果从数据流中读出奇数个数值，那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值，那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流，使用GetMedian()方法获取当前读取数据的中位数。

参考代码

class Solution {
public:
    void Insert(int num)
    {
        int grossSum = mMax.size() + mMin.size();
        if(grossSum & 1){
            if(!mMin.empty() && num > *mMin.begin()){
                int temp = *mMin.begin();
                mMin.erase(mMin.begin());
                mMin.insert(num);
                num = temp;
            }
            mMax.insert(num);
            
        }else{
            if(!mMax.empty() && num < *mMax.begin()){
                int temp = *mMax.begin();
                mMax.erase(mMax.begin());
                mMax.insert(num);
                num = temp;
            }
            mMin.insert(num);
        }
        
    }

    double GetMedian()
    { 
        int grossSum = mMax.size() + mMin.size();
        if(grossSum & 1)
            return (double)(*mMin.begin());
        else
            return (double)(*mMax.begin() + *mMin.begin()) / 2;
    }
    
    
private:
    multiset<int, greater<int>> mMax;
    multiset<int, less<int>> mMin;
    
};

注意：multiset不能通过下标法索引。