牛客——数据流中的中位数

题目描述
如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流,使用GetMedian()方法获取当前读取数据的中位数。
题解:
1,声明一个vector数组
2,在insert函数中每次往vector数组中push_back一个num
3,计算数组长度,奇数则输出中间值,否则输出中间两值的平均值

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

    double GetMedian()
    { 
        sort(array.begin(), array.end());
        int length = array.size();
        if (length%2==0)
            return double((array[length/2]+array[length/2-1]))/2;
        else
            return double(array[length/2]);
    }
};
发布了315 篇原创文章 · 获赞 119 · 访问量 11万+

猜你喜欢

转载自blog.csdn.net/w144215160044/article/details/105004975