数据流中的中位数

题目描述

如何得到一个数据流中的中位数？如果从数据流中读出奇数个数值，那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值，那么中位数就是所有数值排序之后中间两个数的平均值。我们使用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;
}