Find Median from Data Stream - LeetCode

Median is the middle value in an ordered integer list. If the size of the list is even, there is no middle value. So the median is the mean of the two middle value.

Examples:

[2,3,4] , the median is 3

[2,3], the median is (2 + 3) / 2 = 2.5

Design a data structure that supports the following two operations:

• void addNum(int num) - Add a integer number from the data stream to the data structure.
• double findMedian() - Return the median of all elements so far.

For example:

add(1)
add(2)
findMedian() -> 1.5
add(3)
findMedian() -> 2

思路：要获得中位数，要设置两个堆，一个堆存储较小的那一半数，这里构建一个大顶堆small，从而就能获得这一半数中的最大值。

另一个堆为小顶堆large，存储较大的那一半数，堆顶为这一半数中的最小值。

具体实现时，我们将所有的数一个个地先插入到small中，插入后，所有的数重新排序后，堆顶为所有数中的最大值，将其从堆中弹出并插入到large中。这里要判断下small中数量是否比large要少，少的话则要从large中再取回一个数到small中。如果不添加这一步的话，则small中的数会一直添加到large中，最后small为空。这样做能让两堆数数量达到平衡。

在实际实现中，我们用优先队列来实现。C++里的优先队列是大顶堆，因此large在实现时要将所有的数都取负，这样原本的最小值取负后就是现在的最大值了。这也给我们一个提示，如果不想写堆的代码，可以直接用优先队列。

 class MedianFinder {
 public:
     priority_queue<long long> small;
     priority_queue<long long> large;
     // Adds a number into the data structure.
     void addNum(int num) {
         small.push(num);
         large.push(-small.top());
         small.pop();
         if (small.size() < large.size())
         {
             small.push(-large.top());
             large.pop();
         }
     }

     // Returns the median of current data stream
     double findMedian() {
         return small.size() > large.size() ?
             small.top() : (small.top() - large.top()) / 2.0;
     }
 };

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