题目描述
如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流,使用GetMedian()方法获取当前读取数据的中位数。
class Solution {
public:
Solution()
{
count = 0;
}
void Insert(int num)
{
count += 1;
// 元素个数是偶数时,将大顶堆堆顶放入小顶堆
if (count % 2 == 0) {
big_heap.push(num);
small_heap.push(big_heap.top());
big_heap.pop();
}
// 元素个数是奇数时,将小顶堆堆顶放入大顶堆
else {
small_heap.push(num);
big_heap.push(small_heap.top());
small_heap.pop();
}
}
double GetMedian()
{
if (count & 0x1) {
return big_heap.top();
}
else {
return double((small_heap.top() + big_heap.top()) / 2.0);
}
}
private:
int count;
priority_queue<int, vector<int>, less<int>> big_heap; // 左边一个大顶堆
priority_queue<int, vector<int>, greater<int>> small_heap; // 右边一个小顶堆
// 大顶堆所有元素均小于等于小顶堆的所有元素.
};
int main()
{
Solution s;
s.Insert(5);
s.Insert(10);
s.Insert(15);
s.Insert(20);
s.Insert(25);
s.Insert(30);
s.Insert(35);
s.Insert(40);
s.Insert(45);
s.Insert(50);
double num = s.GetMedian();
cout << num << endl;
system("pause");
return 0;
}