【剑指offer】数据流中的中位数

膜拜笔者的引入思路
讨论哪种数据结构存储数据 -> 引出数组 -> 排序的链表 -> 二叉搜索树 -> 优化的AVL树 -> 最后到堆,真是让人深感佩服。

// 以前写代码着重代码的简洁,现在写代码在复杂度相同的情况下,先可读性,再简洁性。

import java.util.PriorityQueue;
import java.util.Queue;

public class Solution {

    Queue<Integer> minHeap = new PriorityQueue<>();
    Queue<Integer> maxHeap = new PriorityQueue<>((o1, o2) -> o2 - o1);
    int count = 0;

    public void Insert(Integer num) {
        if ((++count & 1) == 1) { // 数据总数为奇数放最大堆
            if (!minHeap.isEmpty() && minHeap.peek() < num) {
                int t = minHeap.poll();
                minHeap.add(num);
                maxHeap.add(t);
            } else {
                maxHeap.add(num);
            }

        } else { // 数据总数为偶数放最小堆
            if (maxHeap.peek() > num) {
                int t = maxHeap.poll();
                maxHeap.add(num);
                minHeap.add(t);
            } else {
                minHeap.add(num);
            }
        }
    }

    public Double GetMedian() {
        if ((count & 1) == 1) {
            return (double) maxHeap.peek();
        } else {
            return (maxHeap.peek() + minHeap.peek()) / 2.0;
        }
    }
}
全部评论

相关推荐

05-30 13:04
已编辑
门头沟学院 算法工程师
智谱和米哈游都是ai大模型agent的业务钱的话还是米更多,几乎翻倍了,有没有老哥是两个公司其中一个的,能问问转正率咋样嘛,我问的hr回答都是做的好就可以转正暑期实习
码农索隆:选米哈游:短期高薪、敢承担风险、具备强创新能力,且愿押注游戏AI赛道。 选智谱:稳定性与行业通用能力积累,接受薪资差距以换取更稳妥的职业基础。
投递北京智谱华章科技等公司6个岗位 > 实习期间如何提升留用概率?
点赞 评论 收藏
分享
喜欢核冬天的哈基米很想上市:会爆NullPointerException的
点赞 评论 收藏
分享
Southyeung:我说一下我的看法(有冒犯实属抱歉):(1)简历不太美观,给我一种看都不想看的感觉,感觉字体还是排版问题;(2)numpy就一个基础包,机器学习算法是什么鬼?我感觉你把svm那些写上去都要好一点。(2)课程不要写,没人看,换成获奖经历;(3)项目太少了,至少2-3个,是在不行把网上学习的也写上去。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务