前言

仅记录学习笔记，如有错误欢迎指正。

题目

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

示例：

输入：[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…

解法

利用treeSet自动排序的特性，转为list。

import java.util.*;
public class Solution {
    //tree自动排序
    Set<Integer> set = new TreeSet();
    public void Insert(Integer num) {
    
        set.add(num);
    }

    public Double GetMedian() {
       List<Integer> list = new ArrayList(set);
        int i = list.size()/2;
        //奇数返回中位数
        if(list.size() % 2 == 1){
            return list.get(i)/1.0;
        }else{
            //数据下标从0 开始！
            return (list.get(i)+list.get(i- 1))/2.0;
        }
        
    }


}

0 条评论