牛客刷题之数据流中的中位数

题目描述

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

解题思路：

       首先，将传入的数字存到数组中，然后对数组中的数字进行排序。

       接下来，遍历数组，如果数组中的元素为奇数，则直接返回中间的数字即可；如果是偶数，则返回中间两个数的平均数即可。

import java.util.ArrayList;
import java.util.Collections;
public class Solution {
    ArrayList<Integer> arr = new ArrayList<>();
    public void Insert(Integer num) {
        arr.add(num);
        Collections.sort(arr);
    }

    public Double GetMedian() {
        Double temp = 0.0;
        int len = arr.size();
        for (int i = 0; i < arr.size(); i++) {
            if (arr.size() % 2 != 0) {
                temp = (double) arr.get(len / 2);
            } else {
                temp = (double) (arr.get(len / 2) + arr.get(len / 2 - 1)) / 2;
            }
        }
        return temp;
    }
}