数据流中的中位数

数据流中的中位数

http://www.nowcoder.com/questionTerminal/9be0172896bd43948f8a32fb954e1be1

看大家都是用了优先队列,我一开始没有想到没用了快速排序中partition的思想,也能完成中位数的求解

import java.util.ArrayList;
import java.util.List;
public class Solution {

    private final List<Integer> list = new ArrayList<>();

    public void Insert(Integer num) {
        list.add(num);
    }

    public Double GetMedian() {
        return  partition(list,0,list.size()-1);

    }



    private double partition(List<Integer> list, int lo, int hi) {
        int pivot = list.get(lo);
        int i = lo,j = hi;
        while(i<j){
            while(i<j&&list.get(j)>pivot){
                j--;
            }
            list.set(i,list.get(j));

            while(i<j&&list.get(i)<pivot){
                i++;
            }
            list.set(j,list.get(i));
        }
        list.set(i,pivot);
        if (i<(list.size()>>1)){
            return partition(list,i+1,hi);
        }else if(i>(list.size()>>1)){
            return partition(list,lo,i-1);
        }else {
            if((list.size()%2==1)){
                return list.get(j);
            }else{
                return (list.get(j)+list.get(j-1))*0.5;
            }
        }
    }


}
全部评论

相关推荐

04-29 18:07
常州大学 Java
寂静羽翼:兄弟我已经亲身经历了,双非没实习很多大厂还是会给笔试的,可是有的公司笔试做的好也不给面一直卡着,ssob基本看我没实习都拒绝我了,但是每天投满偶尔也能有一两场初创公司的面试,但是薪资基本在五六千
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务