首页 > 其他 > 详细

295. Find Median from Data Stream

时间:2018-11-25 17:03:57      阅读:145      评论:0      收藏:0      [点我收藏+]

一、题目

  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

原文:https://www.cnblogs.com/skillking/p/10016084.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!