数据流的中位数--剑指Offer(JAVA)

题目:
如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。
思路:
我们选用存放数据的容器是最大堆和最小堆,左侧为最大堆,右侧为最小堆。最大堆的数字全部小于最小堆的数字,且最小堆的尺寸要么与最大堆相同,要么比最大堆尺寸大一。因此,当原本两堆大小相同时,新来的数字首先被加到最大堆,之后最大堆将堆顶数字加给最小堆(先加入最大堆的意义在于,保证传递给最小堆数字后,仍保持两堆大小关系),此时的中位数就是最小堆的堆顶。当最小堆尺寸比最大堆大一时,原步骤不变,之后因为最大堆尺寸比最小堆小2,因此最小堆再将堆顶传给最大堆。此时中位数是两堆和的一半。
代码比较明了

import java.util.*;
//PriorityQueue默认是最小堆,设置最大堆时要重新设置比较器
public class Solution {
    public static PriorityQueue<Integer> minheap, maxheap;
    public Solution(){
        minheap = new PriorityQueue<>();
        maxheap = new PriorityQueue<>(20, new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o2-o1;
            }
        });
    }
    public static void addnum(int n){
        //这里最小堆没有直接加新输入的值,而是先把数值添加到最大堆,再添加最大堆堆顶。保证最小堆新添加的值大于等于最大堆所有值
        maxheap.offer(n);
        minheap.offer(maxheap.poll());
        if (maxheap.size()+1<minheap.size())
        {
            maxheap.offer(minheap.poll());
        }
    }
    public static double GetMedian()
    {
        if (maxheap.size()==0&&minheap.size()==0)
            return -1;
        if (maxheap.size()==0)
            return minheap.peek();
        if (minheap.size()==maxheap.size())
            return (minheap.peek()+maxheap.peek())*0.5;
        else return minheap.peek();
    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值