剑指offer41 数据流中的中位数

题目要求获取一个数据流排序后的中位数，那么可以使用两个优先队列（堆）实现。

本题使用一个大顶堆和一个小顶堆：大顶堆用来存放较小的那一半值，小顶堆用来存放较大的那一半值。输入数组元素中，需要保持两个堆的长度相等，如果数组长度为奇数，就使小顶堆为长度更大的那一个，最后中位数就是小顶堆的堆顶。

如果直接把值插入到堆里，因为无法确定这个值是较大的那一半还是较小的那一半，如果是较大的那一半，直接放入小顶堆不会破坏顺序，反之如果是较小的那一半，会破坏堆内的顺序，所以：

当两个堆长度一样时，先放入大顶堆，大顶堆重新排序后将栈顶放入小顶堆；

当两个堆长度不同时，先放入小顶堆，小顶堆重新排序后将栈顶放入大顶堆。

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();
    }
}