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);
    }
}
相关推荐
好家伙VCC4 分钟前
**发散创新:探索群体智能编程中的新境界**随着科技的飞速发展,群体智能逐渐成为编程领域的一大研究热点。本文将深入探讨群体智能的概念、优
java·python·科技
aramae14 分钟前
数据结构与算法(递归)
开发语言·经验分享·笔记·算法
小欣加油16 分钟前
leetcode 329 矩阵中的最长递增路径
c++·算法·leetcode·矩阵·深度优先·剪枝
秉承初心18 分钟前
Java 23种设计模式的详细解析
java·设计模式
Emilia486.19 分钟前
【Leetcode&nowcode&数据结构】单链表的应用(初阶)
c语言·数据结构·算法·leetcode
千码君201622 分钟前
Go语言:记录一下Go语言系统学习的第一天
java·开发语言·学习·golang·gin·并发编程·编译语言
聪明的笨猪猪24 分钟前
Java 面试清单(含超通俗生活案例与深度理解)
java·经验分享·笔记·面试
MrSYJ42 分钟前
学完涨工资的技巧2:Spring Authorization Server如何签发JWTToken
java·spring boot·微服务
珹洺44 分钟前
Java-Spring入门指南(二十六)Android Studio下载与安装
java·spring·android studio
JAVA学习通1 小时前
JDK高版本特性总结与ZGC实践
java·jvm·算法