295. 数据流的中位数

void1024

于 2024-07-17 00:21:34 发布

阅读量80

点赞数 2

分类专栏：面试经典 150 题文章标签： java 前端服务器

本文链接：https://blog.csdn.net/diOSyu/article/details/140480459

版权

面试经典 150 题专栏收录该内容

132 篇文章 1 订阅

订阅专栏

class MedianFinder {
    PriorityQueue<Integer> fq = new PriorityQueue<>(new Comparator<Integer>() {
        @Override
        public int compare(Integer o1, Integer o2) {
            return o2 - o1;
        }
    });
    PriorityQueue<Integer> sq = new PriorityQueue<>();


    public MedianFinder() {
    }

    public void addNum(int num) {
        if (fq.size() <= sq.size()) {
            sq.add(num);
            fq.add(sq.poll());
        } else {
            fq.add(num);
            sq.add(fq.poll());
        }
    }

    public double findMedian() {
        if (fq.size() == sq.size()) {
            return (fq.peek() + sq.peek()) / 2.0;
        } else {
            return fq.peek();
        }
    }
}