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

题目描述

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

思路

通过insert函数将数据流从小到大排序
通过GetMedian()输出对应中位数

代码

class Solution {
public:
    vector<int> res;
    void Insert(int num){
        int i = 0;
        if(res.size()==0)
        {
            res.push_back(num);
            return;
        }
            
        for(i=0;i<res.size();i++)
        {
            if(num < res[i])
            {
                res.insert(res.begin()+i,num);
                return;
            }
        }
        res.push_back(num);
        return;
    }
    double GetMedian()
    {
        int size = res.size();
        if(size%2 == 0)
        {
            return (res[size/2-1] + res[size/2])/2.0;
        }
        else
            return res[(size-1)/2];
    }
};