其他分享
首页 > 其他分享> > 295. 数据流的中位数

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;
    }
}

标签:count,minQue,double,maxQue,中位数,PriorityQueue,数据流,295
来源: https://blog.csdn.net/weixin_43891573/article/details/115259690