题目描述
如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流,使用GetMedian()方法获取当前读取数据的中位数。
使用TreeMap的自然排序,key值会自动按从小到大排列,然后配合使用TreeMap的higherEntry获取相应的元素。思路简单清晰,本质使用了桶排序的方式。
import java.util.*;
public class Solution {
TreeMap<Integer,Integer> map = new TreeMap<Integer,Integer>();
public void Insert(Integer num) {
// 利用TreeMap的自然排序让插入的元素的key从小到大排序,重复的元素value++统计个数
if(map.containsKey(num)){
int sum = map.get(num);
map.replace(num,sum,sum+1);
}else{
map.put(num,1);
}
}
public Double GetMedian() {
int size = map.size();
int mid = (size+1)/2;
Map.Entry entry = map.firstEntry();
int i = (int)entry.getValue();
while(mid>i){
entry = map.higherEntry((int)entry.getKey()); // 获取最近的一个比当前key大的entry
i+=(int)entry.getValue(); // 累加获取中心点
}
if(size%2==0&& i == mid){
return (((int)entry.getKey()) * 1.0 + (int)map.higherEntry((int)entry.getKey()).getKey()* 1.0) / 2;
}
return (int)entry.getKey()*1.0;
}
}