数据流中的中位数

mmmmmao77

于 2021-12-11 22:07:06 发布

阅读量292

点赞数

CC 4.0 BY-SA版权

分类专栏： java基础文章标签： java

本文链接：https://blog.youkuaiyun.com/dada032/article/details/121880408

java基础专栏收录该内容

30 篇文章

订阅专栏

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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();
 */

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