题目描述
如何得到一个数据流中的中位数？如果从数据流中读出奇数个数值，那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值，那么中位数就是所有数值排序之后中间两个数的平均值。
思路：这道题用到两个堆，使用优先级队列进行实现，原始的优先级队列是最大堆，加上greater仿函数之后是最小堆（加上functional头文件）
一定要保证前面的最大堆元素多于后面的最小堆元素，首先将元素压入最大堆，然后取出最小的元素压入最小堆，然后判断两者大小，最大堆数目小于最小堆，流需要压一个最小堆的元素到前面的最大堆。
最后得结果的时候，是奇数返回前面最大堆的top，否则数目不为0的时候，返回两个堆的top平均数。
class Solution { public: void Insert(int num){ maxHeap.push(num); int tmp = maxHeap.top(); maxHeap.pop(); minHeap.push(tmp); if(maxHeap.size() < minHeap.size()){ maxHeap.push(minHeap.top()); minHeap.pop(); } } double GetMedian(){ if(maxHeap.empty() && minHeap.empty()){ return 0.0; } int sz = maxHeap.size() + minHeap.size(); if(sz % 2 == 0){ return (maxHeap.top() + minHeap.top()) / 2.0; } else{ return maxHeap.top(); } return 0.0; } private: priority_queue<int> maxHeap; priority_queue<int,vector<int>,greater<int>> minHeap; };
