数据流中的中位数

使用优先队列实现中位数查找

最新推荐文章于 2025-12-09 16:25:48 发布

原创最新推荐文章于 2025-12-09 16:25:48 发布 · 303 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java

java基础专栏收录该内容

30 篇文章

订阅专栏

这个博客介绍了一种使用两个优先队列（一个大顶堆和一个小顶堆）来实现中位数查找的数据结构。MedianFinder类通过维护堆的平衡，在添加元素后能快速找到序列的中位数。添加元素时，根据堆的大小调整元素在两个堆中的位置，确保中位数始终可即时获取。

class MedianFinder {

    /** initialize your data structure here. */
    PriorityQueue<Integer> left;
     PriorityQueue<Integer> right;
    public MedianFinder() {
        left=new PriorityQueue<Integer>((x,y)->y-x);
        right = new PriorityQueue<Integer>();
    }
    
    public void addNum(int num) {
        if(left.size()==right.size()){
            right.offer(num);
            left.offer(right.poll());
        }else{
           left.offer(num);
            right.offer(left.poll());
        }
    }
    
    public double findMedian() {
        if(left.size()==0 && right.size()==0){
            return 0 ;
        }

        if(left.size()==right.size()){
            return ((double)left.peek()+right.peek())/2;
        }else{
            return (double) left.peek();
        }

    }
}

/**
 * Your MedianFinder object will be instantiated and called as such:
 * MedianFinder obj = new MedianFinder();
 * obj.addNum(num);
 * double param_2 = obj.findMedian();
 */

使用两个优先队列解决这个问题比较简单，一个大顶堆，一个小顶堆。