题解 | #数据流中的中位数#

数据流中的中位数

http://www.nowcoder.com/practice/9be0172896bd43948f8a32fb954e1be1

描述
如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流,使用GetMedian()方法获取当前读取数据的中位数。
示例1
输入:
[5,2,3,4,1,6,7,0,8]
返回值:
"5.00 3.50 3.00 3.50 3.00 3.50 4.00 3.50 4.00 "
说明:
数据流里面不断吐出的是5,2,3...,则得到的平均数分别为5,(5+2)/2,3...

class Solution {
    vector<int> nums;  // 存储插入的数据
    int N = 0;         // 记录插入数据个数
public:
    void Insert(int num) {
        nums.push_back(num);
        int i = N-1;
        while (i >= 0 && nums[i] > num) {  // 类似于插入排序
            nums[i+1] = nums[i];           // 大于则后移
            --i;
        }
        nums[i+1] = num;
        ++N;
    }

    double GetMedian() { 
        return N%2 ? nums[N/2] : (nums[N/2]+nums[N/2-1])/2.0; // 返回中位数
    }
};
全部评论

相关推荐

Hello_WordN:咱就是说,除了生命其他都是小事,希望面试官平安,希望各位平时也多注意安全
点赞 评论 收藏
分享
爱看电影的杨桃allin春招:我感觉你在炫耀
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务