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

Eran_2023-10-28 9:35

题目

如何得到一个数据流中的中位数？如果从数据流中读出奇数个数值，那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值，那么中位数就是所有数值排序之后中间两个数的平均值。我们使用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 "

解答

源代码

java 复制代码

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;
        }
    }
}

总结

为了便于找出中位数，我们每次在放入数据时都要对当前数据流进行排序，在集合中将数据按照递增序列顺序存储，这样获取中位数时就可以根据集合大小直接找到所需下标的数字，最后得到中位数。