剑指 Offer 41. 数据流中的中位数（困难）

最新推荐文章于 2022-05-07 08:50:10 发布

迪恩_Emma

最新推荐文章于 2022-05-07 08:50:10 发布

阅读量84

点赞数

分类专栏：力扣刷题之剑指offer 力扣刷题之堆文章标签：数据结构 java 优先队列

本文链接：https://blog.csdn.net/di_ko/article/details/116422923

版权

力扣刷题之剑指offer 同时被 2 个专栏收录

70 篇文章 0 订阅

订阅专栏

力扣刷题之堆

3 篇文章 0 订阅

订阅专栏

思路：

使用2个优先队列，一个存储较大的一半，另一个存储较小的一半

如果是奇数，就返回小顶堆（存储较大一半）的堆顶元素

如果是偶数，就返回两堆堆顶元素/2.0的值

代码：

class MedianFinder {
	
	Queue<Integer> A,B;
    /** initialize your data structure here. */
    public MedianFinder() {
		A=new PriorityQueue<>();//小顶堆（存储较大的一半）
		B=new PriorityQueue<>((a,b)->(b-a));//大顶堆（存储较小的一半）
    }
    
    public void addNum(int num) {
		if(A.size()==B.size()){
			//向A中添加元素,必须先将此元素添加到B中确认是否属于较小的一半
			B.add(num);
			A.add(B.poll());
		}
		else{
			//向B中添加元素，必须先将此元素添加到A中确认是否属于较大的一方
			A.add(num);
			B.add(A.poll());
		}
    }
    
    public double findMedian() {
		if(A.size()==B.size()){
			//偶数，取中间2数之和/2
			//因为返回的是double类型，所以要/2.0
			return (A.peek()+B.peek())/2.0;
		}
		else{
			//奇数,取A的顶端元素
			return A.peek();
		}
    }
}

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