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);
    }
}
相关推荐
Miraitowa_cheems1 分钟前
LeetCode算法日记 - Day 88: 环绕字符串中唯一的子字符串
java·数据结构·算法·leetcode·深度优先·动态规划
黑云压城After17 分钟前
vue2实现图片自定义裁剪功能(uniapp)
java·前端·javascript
B站_计算机毕业设计之家44 分钟前
python电商商品评论数据分析可视化系统 爬虫 数据采集 Flask框架 NLP情感分析 LDA主题分析 Bayes评论分类(源码) ✅
大数据·hadoop·爬虫·python·算法·数据分析·1024程序员节
小白菜又菜1 小时前
Leetcode 1518. Water Bottles
算法·leetcode·职场和发展
长存祈月心1 小时前
Rust Option 与 Result深度解析
算法
zcl_19912 小时前
记一次ThreadLocal导致的生产事故
java
RoboWizard2 小时前
怎么判断我的电脑是否支持PCIe 5.0 SSD?Kingston FURY Renegade G5
java·spring·智能手机·电脑·金士顿
杭州杭州杭州2 小时前
机器学习(3)---线性算法,决策树,神经网络,支持向量机
算法·决策树·机器学习
毕设源码-钟学长2 小时前
【开题答辩全过程】以 儿童游泳预约系统为例,包含答辩的问题和答案
java·eclipse
皮皮林5513 小时前
5种接口频率监控方案实战对比,性能、成本、复杂度全解析!
java