随时找到数据流的中位数
【题目】 有一个源源不断地吐出整数的数据流,假设你有足够的空间来 保存吐出的数。请设计一个名叫MedianHolder的结构, MedianHolder可以随时取得之前吐出所有数的中位数。
【要求】
1.如果MedianHolder已经保存了吐出的N个数,那么任意时刻 将一个新数加入到MedianHolder的过程,其时间复杂度是 O(logN)。
2.取得已经吐出的N个数整体的中位数的过程,时间复杂度为 O(1)。
解题思路:
(堆的应用)
1、准备一个大根堆,一个小根堆
2、一个新进来的数,如果比大根堆堆顶大,进小根堆;
比大根堆堆顶小,进大根堆
3、如果两个堆长度差2,弹出堆顶进对面堆
4、最后中位数必在两个堆顶中
小贴士:PriorityQueue优先级队列,实质就是一个栈,默认构成小根堆,如果想实现大根堆的结构,需要自己定义比较器
定义大根堆:
定义小根堆: