如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流,使用GetMedian()方法获取当前读取数据的中位数。
为了保证插入新数据和取中位数的时间效率都高效,这里使用大顶堆+小顶堆的容器,并且满足: 1、两个堆中的数据数目差不能超过1,这样可以使中位数只会出现在两个堆的交接处; 2、大顶堆的所有数据都小于小顶堆,这样就满足了排序要求。 数据排列为: ~~~~~~~~Maxheap minheap~~~~~
为了实现此方法,我们需要平分两个堆,奇数放一个堆,偶数放一个堆里,并且每次存数据时候把堆顶弹到另外一个堆里
public class myComperator implements Comparator<Integer>{
@Override //从大到小排序
public int compare(Integer o1, Integer o2) {
return o2-o1;
}
}
// ~~~~~~~~Maxheap minheap~~~~~
PriorityQueue<Integer> minHeap=new PriorityQueue<>();//根最小
PriorityQueue<Integer> MaxHeap=new PriorityQueue<>(new myComperator());//根最大
int count=0;
public void Insert(Integer num) {
if ((count&1)==0){//偶数,放小根堆
minHeap.offer(num);
MaxHeap.offer(minHeap.poll());
}else {
MaxHeap.offer(num);
minHeap.offer(MaxHeap.poll());
}
count++;
}
public Double GetMedian() {
return (count&1)==0?new Double((minHeap.peek() + MaxHeap.peek())+"")/2:new Double(MaxHeap.peek()+"");
}
import java.util.*;
public class Solution {
ArrayList<Integer> al = new ArrayList<Integer>();
public void Insert(Integer num) {
al.add(num);
Collections.sort(al);
}
public Double GetMedian() {
int mid = al.size()/2;
if((al.size()&1) == 0){
Integer n1 = al.get(mid);
Integer n2 = al.get(mid - 1);
double s = (Double.valueOf(n1 + "") + Double.valueOf(n2 + ""))/2;
return s;
}else{
double s = Double.valueOf(al.get(mid) + "");
return s;
}
}
}