数据流的中心(两个堆,堆插入复杂度最低)

数据流中的中位数

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

/*
堆排序插入O(logn) 得到中位数为O(1);
插入排序插入,折、、半查找为O(logN),移动为O(n) 得到中位数 为O(1)
如果插入数字小于大根堆(存放的是数组小的值,存放数组左边)的最大值(第一个值) 加入
否则加入小根堆(存放数组右边)
*/
class Solution {
public:
    priority_queue<int> left;
    priority_queue<int,vector<int>, greater<int>> right;
    void Insert(int num)
    {
        left.push(num);
        right.push(left.top());
        left.pop();
        if(right.size() > left.size()){ //右边的堆比左边高 平衡
            left.push(right.top());// 且每次只高一个高度
            right.pop();
        }

    }

    double GetMedian()
    { 
        return (left.size() > right.size())? left.top(): (left.top()+right.top())/2.0;
    }

};
全部评论

相关推荐

10-24 11:10
山西大学 Java
若梦难了:哥们,面试挂是很正常的。我大中厂终面挂,加起来快10次了,继续努力吧。
点赞 评论 收藏
分享
shtdbb_:还不错,没有让你做了笔试再挂你
点赞 评论 收藏
分享
1 收藏 评论
分享
牛客网
牛客企业服务