数据流中的中位数

数据流中的中位数

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

用两个堆来实现

  • 如果是偶数个的话,大顶堆用来存放顺序数据流左边的元素,小顶堆用来存放顺序数据流右边的元素,中位数为两个堆顶的平均数。
  • 如果是奇数个的话,大顶堆存放顺序数据流+中位数,小顶堆用来存放顺序数据流右边的元素,中位数为大顶堆的堆顶。
import java.util.*;
public class Solution {

    private static PriorityQueue<Integer> maxHeap= new PriorityQueue<>((o1,o2)->(o2-o1));//左区间,用大顶堆来维护
    private static PriorityQueue<Integer> minHeap= new PriorityQueue<>(); //右区间,用小顶堆来维护
    private static int sum = 0;//用来记录数据流元素的个数(已经插入的个数)
    public static void Insert(Integer num) {
        if(sum %2 == 0) {
            maxHeap.add(num);//当两个堆里的所有元素和是偶数
        }
        else {
            minHeap.add(num);
        }
        //最大堆一定有数据,最小堆不一定有
        //当最大堆中最顶端元素大于最小堆的最顶端元素,则交换两个元素
        if(!minHeap.isEmpty() && maxHeap.peek()>minHeap.peek()) {

            int temp1 = minHeap.poll();
            int temp2 = maxHeap.poll();
            minHeap.add(temp2);
            maxHeap.add(temp1);
        }
        sum++;
    }

    public static Double GetMedian() {
        if(sum %2 == 1) {
            return (double)maxHeap.peek();
        }else {
            return (minHeap.peek() + maxHeap.peek())/2.0;
        }
    }


}
全部评论

相关推荐

来,说点可能被同行“骂”的大实话。🙊当初接数字马力Offer时,朋友都说:“蚂蚁的“内包”公司?你想清楚啊!”但入职快一年后的今天,我反而对他有了不一样的看法!🔹&nbsp;是偏见?还是信息差!之前没入职之前外面都在说什么岗位低人一等这类。实际上:这种情况不可至否,不能保证每个团队都是其乐融融。但我在的部门以及我了解的周边同事都还是十分好相处的~和蚂蚁师兄师姐之间也经常开一些小玩笑。总之:身份是蚂蚁公司给的,地位是自己挣的(一个傲娇女孩的自述)。🔹&nbsp;待遇?玩的就是真实!试用期工资全额发!六点下班跑得快(早9晚6或者早10晚7,动态打卡),公积金顶格交。别听那些画饼的,到手的钱和下班的时间才是真的(都是牛马何必难为牛马)。🔹&nbsp;能不能学到技术?来了就“后悔”!我们拥有权限直通蚂蚁知识库,技术栈多到学不完。说“学不到东西”的人,来了可能后悔——后悔来晚了(哈哈哈哈,可以不学但是不能没有)!💥&nbsp;内推地址:https://app.mokahr.com/su/ueoyhg❗我的内推码:NTA6Nvs走我的内推,可以直达业务部门,面试流程更快速,进度可查!今天新放HC,之前挂过也能再战!秋招已经正式开始啦~机会就摆在这,敢不敢来试一试呢?(和我一样,做个勇敢的女孩)
下午吃泡馍:数字马力的薪资一般哇,5年经验的java/测试就给人一万出头,而且刚入职第三天就让人出差,而且是出半年
帮你内推|数字马力 校招
点赞 评论 收藏
分享
程序员牛肉:你这简历有啥值得拷打的?在牛客你这种简历一抓一大把,也就是个人信息不一样而已。 关键要去找亮点,亮点啊,整个简历都跟流水线生产出来的一样。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务