题目描述:
如何得到一个数据流中的中位数?
如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。
如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。
样例
输入:1, 2, 3, 4
输出:1,1.5,2,2.5
解释:每当数据流读入一个数据,就进行一次判断并输出当前的中位数。
分析:
维护一个大根堆和一个小根堆,大根堆存储较小的那部分数,小根堆存储较大的那部分数。
对于插入操作:先统一插入大根堆,如果大根堆堆顶元素大于小根堆堆顶元素,则交换两个堆顶。当大根堆元素个数比小根堆元素个数多2时,要将大根堆堆顶元素取出插入到小根堆中,以防止两个堆元素个数差大于1.
对于读取中位数操作:若两个堆元素一样多,则输出堆顶的平均值即可,否则输出大根堆堆顶元素。
class Solution {
public:
priority_queue<int> maxh;
priority_queue<int,vector<int>,greater<int> > minh;
void insert(int num){
maxh.push(num);
if(minh.size() && minh.top() < maxh.top()){
int x = minh.top(),y = maxh.top();
minh.pop(),maxh.pop();
minh.push(y),maxh.push(x);
}
if(maxh.size() == minh.size() + 2){
int x = maxh.top();
maxh.pop();
minh.push(x);
}
}
double getMedian(){
if(maxh.size() == minh.size()) return (maxh.top() + minh.top()) / 2.0;
return maxh.top();
}
};