数据流中的中位数
题目描述:
如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。
思路:
代码:
class Solution {
public:
void Insert(int num)
{
int size = minHeap.size()+maxHeap.size();
if ((size&1)==0) //偶数
{
//若比大顶堆最大元素大,则先插入到小顶堆中再从中取最小值
if (!maxHeap.empty()&&num>maxHeap.top())
{
minHeap.push(num);
num = minHeap.top();
minHeap.pop();
}
maxHeap.push(num);
}
else
{ //若比大顶堆最大元素小,则先插入到大顶堆中再从中取最大值
if (num<maxHeap.top())
{
maxHeap.push(num);
num = maxHeap.top();
maxHeap.pop();
}
minHeap.push(num);
}
}
double GetMedian()
{
int size = minHeap.size()+maxHeap.size();
if (size&1)
{
return maxHeap.top();
}
return (double(maxHeap.top()+minHeap.top()))/2;
}
private:
priority_queue<int,vector<int>,greater<int>> minHeap;
priority_queue<int> maxHeap;
};