295. 数据流的中位数

中位数是有序列表中间的数。如果列表长度是偶数,中位数则是中间两个数的平均值。
例如,
[2,3,4] 的中位数是 3
[2,3] 的中位数是 (2 + 3) / 2 = 2.5
设计一个支持以下两种操作的数据结构:
void addNum(int num) - 从数据流中添加一个整数到数据结构中。
double findMedian() - 返回目前所有元素的中位数。

/**
    优先队列
*/
 
class MedianFinder {
 
    private int count;
    private PriorityQueue<Integer> maxQue;
    private PriorityQueue<Integer> minQue;
 
    /** initialize your data structure here. */
    public MedianFinder() {
        count = 0;
        maxQue = new PriorityQueue<>((x1,x2)->x2-x1);
        minQue = new PriorityQueue<>();
    }
    
    public void addNum(int num) {
        //计数器
        count++;
        //加入大根堆
        maxQue.offer(num);
        //将大根堆的最大放入小根堆
        minQue.offer(maxQue.poll());
        //总数为奇数,就把小根堆最小放入大根堆
        if( (count & 1) != 0 )
            maxQue.offer(minQue.poll());
    }
    
    public double findMedian() {
        if((count&1)!=0)
            return (double)maxQue.peek();
        else
            return (double)(maxQue.peek()+minQue.peek())/2;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值