力扣HOT100 - 295. 数据流的中位数

最新推荐文章于 2024-07-21 23:31:38 发布

_OLi_

最新推荐文章于 2024-07-21 23:31:38 发布

阅读量223

点赞数 2

分类专栏：力扣HOT100 算法题合集文章标签： leetcode 算法 java

本文链接：https://blog.csdn.net/qq_61504864/article/details/138759382

版权

力扣HOT100 算法题合集专栏收录该内容

100 篇文章 0 订阅

订阅专栏

解题思路：

小顶堆 + 大顶堆

class MedianFinder {
    Queue<Integer> A, B;

    public MedianFinder() {
        A = new PriorityQueue<>();
        B = new PriorityQueue<>((x, y) -> (y - x));
    }

    public void addNum(int num) {
        if (A.size() != B.size()) {
            A.add(num);
            B.add(A.poll());
        } else {
            B.add(num);
            A.add(B.poll());
        }
    }

    public double findMedian() {
        return A.size() != B.size() ? A.peek() : (A.peek() + B.peek()) / 2.0;
    }
}