295. 数据流的中位数

295. 数据流的中位数
中位数是有序整数列表中的中间值。如果列表的大小是偶数,则没有中间值,中位数是两个中间值的平均值。

例如 arr = [2,3,4] 的中位数是 3 。
例如 arr = [2,3] 的中位数是 (2 + 3) / 2 = 2.5 。
实现 MedianFinder 类:

MedianFinder() 初始化 MedianFinder 对象。

void addNum(int num) 将数据流中的整数 num 添加到数据结构中。

double findMedian() 返回到目前为止所有元素的中位数。与实际答案相差 10-5 以内的答案将被接受。

list 存储 超时

java 复制代码
class MedianFinder {

    List<Integer> list = new ArrayList<>();

    public MedianFinder() {

    }
    
    public void addNum(int num) {
        int i = list.size();
        while (i > 0) {
            if (num < list.get(i-1)) {
                i-=1;
            } else {
                break;
            }
        }
        if (i == list.size()) {
            list.add(num);
        } else {
            list.add(i, num);
        }
    }
    
    public double findMedian() {
        int l = list.size();
        if (l % 2 == 1) {
            return list.get(l/2);
        }
        return (list.get(l/2) + list.get(l/2-1)) / 2.0D;
    }
}

/**
 * Your MedianFinder object will be instantiated and called as such:
 * MedianFinder obj = new MedianFinder();
 * obj.addNum(num);
 * double param_2 = obj.findMedian();
 */

最大最小堆

java 复制代码
class MedianFinder {

    PriorityQueue<Integer> maxHeap = new PriorityQueue<Integer>((a, b) -> b-a);
    PriorityQueue<Integer> minHeap = new PriorityQueue<Integer>((a, b) -> a-b);

    public MedianFinder() {

    }
    
    public void addNum(int num) {
        if (maxHeap.isEmpty() || num < maxHeap.peek()) {
            maxHeap.offer(num);
            if (minHeap.size() + 1 < maxHeap.size()){
                minHeap.offer(maxHeap.poll());
            }
        }  else {
            minHeap.offer(num);
            if (minHeap.size() > maxHeap.size()) {
                maxHeap.offer(minHeap.poll());
            }
        }
    }
    
    public double findMedian() {
        if (maxHeap.size() > minHeap.size()) {
            return maxHeap.peek();
        }
        return (minHeap.peek() + maxHeap.peek()) / 2.0D;
    }
}

/**
 * Your MedianFinder object will be instantiated and called as such:
 * MedianFinder obj = new MedianFinder();
 * obj.addNum(num);
 * double param_2 = obj.findMedian();
 */
相关推荐
weixin_462428477 分钟前
使用 Caffeine 缓存并在业务方法上通过注解实现每3到5秒更新缓存
java·缓存
程序媛小果9 分钟前
基于java+SpringBoot+Vue的桂林旅游景点导游平台设计与实现
java·vue.js·spring boot
骑鱼过海的猫12311 分钟前
【java】java通过s3访问ceph报错
java·ceph·iphone
杨充17 分钟前
13.观察者模式设计思想
java·redis·观察者模式
Lizhihao_19 分钟前
JAVA-队列
java·开发语言
喵叔哟28 分钟前
重构代码之移动字段
java·数据库·重构
喵叔哟28 分钟前
重构代码之取消临时字段
java·前端·重构
fa_lsyk31 分钟前
maven环境搭建
java·maven
林开落L36 分钟前
前缀和算法习题篇(上)
c++·算法·leetcode
Daniel 大东1 小时前
idea 解决缓存损坏问题
java·缓存·intellij-idea