leetcode295. 数据流的中位数

java 复制代码
class MedianFinder {
    //A为小根堆,B为大根堆
    List<Integer> A,B;
    public MedianFinder() {
        A = new ArrayList<Integer>();
        B = new ArrayList<Integer>();
    }
    
    public void addNum(int num) {
        int m = A.size(),n = B.size();
        if(m == n){
            insert(B,num);
            int top = deleteTop(B);
            insert(A,top);
        }else{
            insert(A,num);
            int top = deleteTop(A);
            insert(B,top);
        }
    }

    //删除堆顶元素并返回其值
    private int deleteTop(List<Integer> list){
        Collections.swap(list,0,list.size() - 1);
        int heapSize = list.size() - 1;
        int root = 0;
        while(root < heapSize/2){
            int left = root * 2 + 1,right = root * 2 + 2,largest = root;
            if(left < heapSize && comparee(list,left,largest))
                largest = left;
            if(right < heapSize &&  comparee(list,right,largest))
                largest = right;
            if(root == largest)
                break;
            Collections.swap(list,root,largest);
            root = largest;
        }
        return list.remove(list.size() - 1);
    }

    //A小根堆的比较方法:比较值<目标值?
    private boolean comparee(List<Integer> list,int source,int target){
        if(list == A)
            return list.get(source) < list.get(target);
        else
            return list.get(source) > list.get(target);
    }

    //将数插入到堆中
    private void insert(List<Integer> list,int num){
        list.add(num);
        int index = list.size() - 1;
        while(index > 0){
            int root = (index - 1) / 2;
            if(!comparee(list,index,root))
                break;
            Collections.swap(list,root,index);
            index = root;
        }
    }
    
    public double findMedian() {
        int m = A.size(),n = B.size();
        return (m==n) ? (A.get(0) + B.get(0)) / 2.0 : A.get(0);
    }
}
相关推荐
卡尔特斯26 分钟前
Android Kotlin 项目代理配置【详细步骤(可选)】
android·java·kotlin
白鲸开源26 分钟前
Ubuntu 22 下 DolphinScheduler 3.x 伪集群部署实录
java·ubuntu·开源
ytadpole34 分钟前
Java 25 新特性 更简洁、更高效、更现代
java·后端
纪莫1 小时前
A公司一面:类加载的过程是怎么样的? 双亲委派的优点和缺点? 产生fullGC的情况有哪些? spring的动态代理有哪些?区别是什么? 如何排查CPU使用率过高?
java·java面试⑧股
JavaGuide2 小时前
JDK 25(长期支持版) 发布,新特性解读!
java·后端
用户3721574261352 小时前
Java 轻松批量替换 Word 文档文字内容
java
白鲸开源2 小时前
教你数分钟内创建并运行一个 DolphinScheduler Workflow!
java
CoovallyAIHub2 小时前
中科大DSAI Lab团队多篇论文入选ICCV 2025,推动三维视觉与泛化感知技术突破
深度学习·算法·计算机视觉
Java中文社群2 小时前
有点意思!Java8后最有用新特性排行榜!
java·后端·面试
代码匠心3 小时前
从零开始学Flink:数据源
java·大数据·后端·flink