剑指 Offer 41. 数据流中的中位数
  1BnnW8rtw7M9 2023年11月02日 38 0


剑指 Offer 41. 数据流中的中位数

class MedianFinder {
    PriorityQueue<Integer> pq1, pq2;

    /** initialize your data structure here. */
    public MedianFinder() {
        pq1 = new PriorityQueue<>((a, b) -> Integer.compare(a, b));  // 返回最小的元素,存放较大的一半
        pq2 = new PriorityQueue<>((a, b) -> Integer.compare(b, a));  // 返回最大的元素,存放较小的一半
    }
    
    public void addNum(int num) {
        if(pq1.size() == pq2.size()){
            pq1.offer(num);
            pq2.offer(pq1.poll());
        }else{
            pq2.offer(num);
            pq1.offer(pq2.poll());
        }
    }
    
    public double findMedian() {
        if(pq1.size() == pq2.size()){
            return (pq1.peek() + pq2.peek()) / 2.0;
        }else{
            return pq2.peek();
        }
    }
}


【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

  1. 分享:
最后一次编辑于 2023年11月08日 0

暂无评论

推荐阅读
1BnnW8rtw7M9
作者其他文章 更多