295. Find Median from Data Stream

public class MedianFinder {
    private Queue small = new PriorityQueue(),
                        large = new PriorityQueue();
    public void addNum(int num) {
        large.add((long) num);
        small.add(-large.poll());
        if (large.size() < small.size())
            large.add(-small.poll());
    }

    public double findMedian() {
        return large.size() > small.size()
               ? large.peek()
               : (large.peek() - small.peek()) / 2.0;
    }
};

// Your MedianFinder object will be instantiated and called as such:
// MedianFinder mf = new MedianFinder();
// mf.addNum(1);
// mf.findMedian();

你可能感兴趣的:(295. Find Median from Data Stream)