题目要求获取一个数据流排序后的中位数,那么可以使用两个优先队列(堆)实现。
本题使用一个大顶堆和一个小顶堆:大顶堆用来存放较小的那一半值,小顶堆用来存放较大的那一半值。输入数组元素中,需要保持两个堆的长度相等,如果数组长度为奇数,就使小顶堆为长度更大的那一个,最后中位数就是小顶堆的堆顶。
如果直接把值插入到堆里,因为无法确定这个值是较大的那一半还是较小的那一半,如果是较大的那一半,直接放入小顶堆不会破坏顺序,反之如果是较小的那一半,会破坏堆内的顺序,所以:
当两个堆长度一样时,先放入大顶堆,大顶堆重新排序后将栈顶放入小顶堆;
当两个堆长度不同时,先放入小顶堆,小顶堆重新排序后将栈顶放入大顶堆。
class MedianFinder {
/** initialize your data structure here. */
Queue<Integer> A,B;
public MedianFinder(){
A=new PriorityQueue<>();
B=new PriorityQueue<>((x,y)->(y-x));
}
public void addNum(int num){
if(A.size()!=B.size()){
A.add(num);
B.add(A.poll());
}
else{
B.add(num);
A.add(B.poll());
}
}
public double findMedian(){
return A.size()==B.size()?(A.peek()+B.peek())/2.0:A.peek();
}
}