数据流中的中位数
题目描述
如何得到一个数据流中的中位数?如果从数据流中读出奇数的个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序后中间两个数的平均值。
思路分析
用一个最大堆实现左边的数据容器,用一个最小堆实现右边的数据容器。首先要保证数据平均分配到两个堆中,因此两个堆中数据的数目之差不能小于1.所以在数据的总数目是偶数时将新数据插入最小堆,否则插入最大堆。代码如下:
import java.util.Comparator;
import java.util.PriorityQueue;
public class Solution {
PriorityQueue<Integer> minHeap=new PriorityQueue<>();
PriorityQueue<Integer> maxHeap=new PriorityQueue<>(20,new Comparator<Integer>(){
@Override
public int compare(Integer o1, Integer o2) {
return o2-o1;
}
});
private int count=0;
public void Insert(Integer num) {
count++;
if(count%2==0){
if(!maxHeap.isEmpty()&&maxHeap.peek()>num){
maxHeap.add(num);
num=maxHeap.poll();
}
minHeap.add(num);
}else{
if(!minHeap.isEmpty()&&minHeap.peek()<num){
minHeap.add(num);
num = minHeap.poll();
}
maxHeap.add(num);
}
}
public Double GetMedian() {
if(maxHeap.size()==minHeap.size()){
return (maxHeap.peek()+minHeap.peek())/2.0;
}else if(maxHeap.size()>minHeap.size()){
return maxHeap.peek()*1.0;
}else {
return minHeap.peek()*1.0;
}
}
}