数据流中的中位数

剑值Offer 面试题41：数据流中的中位数

题目：如何得到一个数据流中的中位数？如果从数据流中读出奇数个数值，那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值，那么中位数就是所有数值排序之后中间两个数的平均值。

代码实现：

template<typename T> class DynamicArray
public:
	
	void Insert(T num){
		
		if(((max.size() + min.size()) & 1) == 0){ //数据的总数目是偶数时，把新数据num插入最小堆
			
			if(max.size() > 0 && num < max[0]){ //如果num比最大堆中的一些数据要小
				
				max.push_back(num); //先把num插入最大堆
				
				push_heap(max.begin(), max.end(), less<T>()); //更新重建最大堆
				
				num = max[0]; //令num为重建最大堆后堆中最大的元素，即max[0]

				pop_heap(max.begin(), max.end(), less<T>()); //pop_heap()是在堆的基础上，弹出最大堆的堆顶元素。
				//需要注意的是，pop_heap()并没有删除元素，而是将堆顶元素和数组最后一个元素进行了替换，
				//如果要删除这个元素，还需要对数组进行pop_back()操作。
				max.pop_back();
			}
			min.push_back(num);  //把num插入到最小堆
			push_heap(min.begin(), min.end(), greater<T>());  //更新重建最小堆
		}
		else{ //数据的总数目是奇数，把新数据num插入最大堆
			if(min.size() > 0 && num > min[0]){ //如果num比最小堆中的一些数据要大
				min.push_back(num); //先把num插入最小堆
				push_heap(min.begin(), min.end(), greater<T>()); //更新重建最小堆

				num = min[0]; //令num为重建最小堆后堆中最小的元素，即min[0]

				//弹出最小堆的堆顶元素
				pop_heap(min.begin(), min.end(), greater<T>()); 
				min.pop_back();
			}
			max.push_back(num); //把num插入到最大堆
			push_heap(max.begin(), max.end(), less<T>()); //更新重建最大堆
		}
	}

	T GetMedian(){ //用来得到容器中所有数据的中位数
		int size = max.size() + min.size();
		if(size == 0)
			throw exception("No numbers are available");

		T median = 0;
		if((size & 1) == 1) //数据总数目为奇数
			median = min[0]; //因为数据的总数目是偶数时是把数据插入到最小堆，即奇数个数据的中位数在最小堆
		else
			median = (min[0] + max[0]) / 2;

		return median;
	}

private:
	vector<T> min;
	vector<T> max;
};