leetcode-295-数据流的中位数字节跳动面试题

hmyqwe

于 2021-03-04 17:09:44 发布

阅读量175

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构文章标签： leetcode

本文链接：https://blog.csdn.net/hmyqwe/article/details/114372942

数据结构专栏收录该内容

11 篇文章

订阅专栏

本文介绍了一种用于处理连续输入整数流并能够高效找到当前所有数值中位数的数据结构实现。通过使用两个堆（一个小顶堆和一个大顶堆），该结构能够在O(log n)的时间复杂度内完成中位数查找。

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

在这里插入图片描述

class MedianFinder {
    
    private PriorityQueue<Integer> large;
    private PriorityQueue<Integer> small;

    /** initialize your data structure here. */
    public MedianFinder() {
        // 小顶堆
        large = new PriorityQueue<>();
        // 大顶堆
        small = new PriorityQueue<>((a, b) -> {
            return b - a;
        });
    }
    
    public void addNum(int num) {
        if (small.size() >= large.size()) {
            small.offer(num);
            large.offer(small.poll());
        } else {
            large.offer(num);
            small.offer(large.poll());
        }
    }
    
    public double findMedian() {
        // 总元素个数为偶数，两个堆的元素不一样多，多的那个堆的堆顶元素就是中位数
        if (large.size() < small.size()) {
            return small.peek();
        } else if (large.size() > small.size()) {
            return large.peek();
        }
        // 总元素个数为奇数，两个堆的元素一样多，两个堆堆顶元素的平均数是中位数
        return (large.peek() + small.peek()) / 2.0;
    }
}

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