题目描述
如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流,使用GetMedian()方法获取当前读取数据的中位数。
思路:
- 这题其实不难,看清楚题意就是让你写一个数据集合,什么容器都可以,然后再取得中位数,
- 第一反应是TreeSet,它是有序的插入后可以有序取出,到时候访问中间的个数就好了,这里就没有写出来
- 我用LinkedList方便插入和查找,然后插入方法就写个排序的头插法,读取中位数太简单了,这里就不讲了,看代码即可
import java.util.LinkedList;
public class Solution {
public LinkedList<Integer> list=new LinkedList<Integer>();
public void Insert(Integer num) {
//插入的数比头小的话往头插
if(list.size()==0|| num < list.getFirst()){
list.addFirst(num);
}else{//比头大的话就要
boolean flag=false;
for(Integer e:list){
if(num<e){
int index=list.indexOf(e);
list.add(index,num);
flag=true;
break;
}
}
if(!flag){ //找不到比他大的数字
list.addLast(num);
}
}
}
public Double GetMedian() {
if(list.size()==0) return null;
if(list.size()%2==0){ //如果个数为偶数个就是中间2个之和除2
int i=list.size()/2;
Double a=Double.valueOf(list.get(i)+list.get(i-1));
return a/2;
}
list.get(0);
Double b = Double.valueOf(list.get((list.size() + 1) / 2 - 1));
return b;
}
}