295. Find Median from Data Stream

Posted skillking

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了295. Find Median from Data Stream相关的知识,希望对你有一定的参考价值。

一、题目

  1、审题

  技术分享图片

  2、分析

    实现一个数据结构,可以添加整形元素,并可以返回排序后的中位数。

 

二、解答

  1、思路

    采用两个 PriorityQueue。

    ①、采用两个 PriorityQueue,PriorityQueue 具有对元素进行自动排序的功能。

    ②、一个为 maxQueue,记录比中位数大的所有元素,另一个为 smallQueue, 记录比中位数小的元素。

    ③、为了方便统计数中位数,smallQueue 中存储的元素为其负值,这样,候补中位数就在 smallQueue 的队头。

    ④、若元素总数为奇数,直接返回 largeQueue 的队头,若元素为偶数,返回 largeQueue 的队头和 smallQueue 队头的负值,两元素的平均。

public class MedianFinder {
    
    /** initialize your data structure here. */
    private Queue<Long> small = new PriorityQueue<Long>();
    private Queue<Long> large = new PriorityQueue<>();
    
    public void addNum(int num) {
        large.add((long) num);
        small.add(-large.poll());
        if(large.size() < small.size())
            large.add(-small.poll());
    }
    
    public double findMedian() {

        if(large.size() > small.size())
            return large.peek();
        
        return (large.peek() - small.peek()) / 2.0;
    }
}

 

以上是关于295. Find Median from Data Stream的主要内容,如果未能解决你的问题,请参考以下文章