295. Find Median from Data Stream

Median is the middle value in an ordered integer list. If the size of the list is even, there is no middle value. So the median is the mean of the two middle value.

Examples:

[2,3,4] , the median is 3

[2,3], the median is (2 + 3) / 2 = 2.5

Design a data structure that supports the following two operations:

  • void addNum(int num) - Add a integer number from the data stream to the data structure.
  • double findMedian() - Return the median of all elements so far.
    For example:
addNum(1)
addNum(2)
findMedian() -> 1.5
addNum(3) 
findMedian() -> 2

一刷
题解:
构造一个maxHeap, minHeap,
maxHeap与minHeap的size相同或者大1
maxHeap里面的值都小于minHeap

class MedianFinder {
    
    private Queue<Integer> maxHeap;
    private Queue<Integer> minHeap;

    /** initialize your data structure here. */
    public MedianFinder() {
        maxHeap = new PriorityQueue<Integer>(10, Collections.reverseOrder());
        minHeap = new PriorityQueue<Integer>(); 
        //maxHeap size == minHeap size (+1)
    }
    
    public void addNum(int num) {
        if(maxHeap.size() == minHeap.size()){
            if(!minHeap.isEmpty() && minHeap.peek()<num){
                maxHeap.offer(minHeap.poll());
                minHeap.offer(num);
            }else{
                maxHeap.offer(num);
            }
        }else{
           if(num<maxHeap.peek()){
               minHeap.offer(maxHeap.poll());
               maxHeap.offer(num);
           }else{
               minHeap.offer(num);
           }
        }
    }
    
    public double findMedian() {
        if(minHeap.size() == maxHeap.size()) {
            double a = maxHeap.peek();
            double b = minHeap.peek();
            return (a + b)/2;
        }
        else return maxHeap.peek();
    }
}

/**
 * Your MedianFinder object will be instantiated and called as such:
 * MedianFinder obj = new MedianFinder();
 * obj.addNum(num);
 * double param_2 = obj.findMedian();
 */
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...
    土汪阅读 12,767评论 0 33
  • 脸凉凉的,原来不知什么时候,已流了两行泪。擦去泪,对自己勉强笑了笑,忽然觉得委屈,眼又汪满了水。仰头看了会子星。心...
    铭玥咏全阅读 297评论 0 1
  • 欠錢不要緊, 要努力賺錢來還, 暫時還不上要致電讓人知道你没忘, 堂堂正正做人,明明白白做事, 不要弄丢了别人對你...
    霖靂盡致阅读 180评论 2 1