如何得到一个数据流中的中位数?

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

class Solution {
public:
    vector<int> result;
    void Insert(int num)
    {
        result.push_back(num);
    }

    double GetMedian()
    { 
        double mid;
        sort(result.begin(),result.end());
        int len = result.size();
        if(len <= 0)
            return 0;
        if(len % 2 == 0)
        {
            mid = ((double)result[len/2-1] + (double)result[len/2]) / 2;
        }
        if(len % 2 == 1)
        {
            mid = (double)result[len/2];
        }
        return mid;
    }
};

思路:其实第一次编译的时候,结果是一个数组,我觉得很奇怪,结果不是应该是只有一个数吗?最后发现这道题其实是一个数组,数组的第一个数找一次中位数,然后前两位数找一次中位数,前3位数找一次,以此类推。

注意事项:这道题的GetMedian()函数中,他的返回类型是double,过程中需要做类型转换,转为double型。

猜你喜欢

转载自blog.csdn.net/daiyun_102474/article/details/81609337