数据流中的中位数
如何得到一个数据流中的中位数?
如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。
如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。
样例
输入: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;
}
};