数据流中的中位数
如何得到一个数据流中的中位数?
如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。
如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。
样例
输入:1, 2, 3, 4 输出:1,1.5,2,2.5 解释:每当数据流读入一个数据,就进行一次判断并输出当前的中位数。
class Solution { public: //大根堆储存较小的数,小根堆储存较大的数,中间的数就是中位数; priority_queue<int> maxq; priority_queue<int,vector<int>,greater<int> > minq; void insert(int num){ maxq.push(num); while(minq.size() && minq.top() < maxq.top()){ int maxv = maxq.top(),minv = minq.top(); maxq.pop();minq.pop(); maxq.push(minv); minq.push(maxv); } if( maxq.size() > minq.size() + 1){ minq.push(maxq.top()); maxq.pop(); } } double getMedian(){ if(minq.size() + maxq.size() & 1) return maxq.top(); else return (minq.top()+maxq.top())/2.0; } };