当前位置：首页 > article >正文

【剑指Offer】41.数据流中的中位数

article 2024/9/27 7:04:27

题目

如何得到一个数据流中的中位数？如果从数据流中读出奇数个数值，那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值，那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流，使用GetMedian()方法获取当前读取数据的中位数。

数据范围：数据流中数个数满足 1≤n≤1000 ，大小满足 1≤val≤1000
进阶：空间复杂度O(n) ，时间复杂度O(nlogn)

示例1

输入：[5,2,3,4,1,6,7,0,8]

返回值："5.00 3.50 3.00 3.50 3.00 3.50 4.00 3.50 4.00 "

说明：数据流里面不断吐出的是5,2,3...,则得到的平均数分别为5,(5+2)/2,3...

示例2

输入：[1,1,1]

返回值："1.00 1.00 1.00 "

解答

源代码

import java.util.*;


public class Solution {
    List<Integer> val = new ArrayList<>();

    public void Insert(Integer num) {
        if (val.isEmpty()) {
            val.add(num);
        } else {
            int i;
            for (i = 0; i < val.size(); i++) {
                if (num < val.get(i)) {
                    break;
                }
            }
            val.add(i, num);
        }
    }

    public Double GetMedian() {
        int len = val.size();

        if (len % 2 == 1) {
            return (double)val.get(len / 2);
        } else {
            int left = val.get(len / 2 - 1);
            int right = val.get(len / 2);
            return (double)(left + right) / 2;
        }
    }
}