41、数据流中的中位数

思路:
前一半数用最大堆保存,后一半数用最小堆保存,并且最大堆的所有数小于最小堆的所有数,这样求中位数就是当最大堆最小堆堆顶的和的一半。

import java.util.ArrayList;
public class Solution {
    ArrayList<Integer> max = new ArrayList<Integer>();
    ArrayList<Integer> min = new ArrayList<Integer>();
    //插入数据流来的数
    public void Insert(Integer num) {
        if(max.size()==0)
            max.add(0);
        if(min.size()==0)
            min.add(0);
        //看是向哪个堆添加数据
        if(max.size()==min.size())
        {
            if((min.size()==1)||(num<=min.get(1)))
                insertMax(max,num);
            else{
                insertMax(max,min.get(1));
                min.remove(1);
                insertMin(min,num);
            }
        }
        else{
            if((max.size()==1)||(num>=max.get(1)))
                insertMin(min,num);
            else{
                insertMin(min,max.get(1));
                max.remove(1);
                insertMax(max,num);
            }
        }
    }
    //得到中位数
    public Double GetMedian() {
        if((max.size()==1)&&(min.size()==1))
            return 0.0;
        else if((max.size()+min.size())%2==1)
            return (double)max.get(1);
        else
            return ((double)(max.get(1)+min.get(1)))/2.0;
    }
    //交换两数
    public void swap(ArrayList<Integer> a,int p1,int p2)
    {
        int temp = a.get(p1);
        a.set(p1,a.get(p2));
        a.set(p2,temp);
    }
    //在最大堆中添加数
    public void insertMax(ArrayList<Integer> max,Integer num)
    {

        int i = max.size();
        max.add(num);
        while((i>1)&&(max.get(i)>max.get(i/2)))
        {
              swap(max,i,i/2);
              i=i/2;
        }
    }
    //在最小堆中添加数
    public void insertMin(ArrayList<Integer> min,Integer num)
    {

        int i = min.size();
        min.add(num);
        while((i>1)&&(min.get(i)<min.get(i/2)))
        {
              swap(min,i,i/2);
              i=i/2;
        }
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值