中位数是有序列表中间的数。如果列表长度是偶数，中位数则是中间两个数的平均值。（LeetCode 295）

最新推荐文章于 2021-02-03 14:10:58 发布

BGnv5

最新推荐文章于 2021-02-03 14:10:58 发布

阅读量882

点赞数

分类专栏：算法文章标签： LeetCode

本文链接：https://blog.csdn.net/qq_44819993/article/details/101636308

版权

算法专栏收录该内容

3 篇文章 0 订阅

订阅专栏

/**

*思路：设置一个大顶堆和一个小顶堆，如果奇数个返回大顶堆堆顶，偶数个返回（大顶堆+小顶堆）/2

*其中大顶堆的个数等于小顶堆的个数或者比小顶堆多一个

*大顶堆中的最大数小于小顶堆的最小数

*如果是奇数个数，大顶堆->小顶堆->大顶堆

*偶数个，大顶堆->小顶堆

class MedianFinder {

        private int count;
        private PriorityQueue<Integer> maxHeap;
        private PriorityQueue<Integer> minHeap;
    
    public MedianFinder() {
        count=0;
        maxHeap= new PriorityQueue<>((x,y)->y-x);
        minHeap = new PriorityQueue<>();
    }
    
    public void addNum(int num) {
        count++;
        maxHeap.offer(num);
        minHeap.add(maxHeap.poll());
        if((count & 1) != 0){
            maxHeap.add(minHeap.poll());
        }
        
    }
    
    public double findMedian() {
        if((count & 1) == 0){
            return (double)(maxHeap.peek()+minHeap.peek())/2;
        }else{
            return maxHeap.peek();
        }
    }
}