Leetcode题目:https://leetcode.com/problems/find-median-from-data-stream/description/
Median is the middle value in an ordered integer list. If the size of the list is even, there is no middle value. So the median is the mean of the two middle value.
Examples:[2,3,4]
, the median is 3
[2,3]
, the median is (2 + 3) / 2 = 2.5
Design a data structure that supports the following two operations:
- void addNum(int num) - Add a integer number from the data stream to the data structure.
- double findMedian() - Return the median of all elements so far.
For example:
addNum(1) addNum(2) findMedian() -> 1.5 addNum(3) findMedian() -> 2
思路
维护一个大根堆和一个小根堆。大根堆存放到目前为止较小的一半数,小根堆存放较大的一半数。这样中位数只可能是堆顶或者堆顶两个数的均值。维护两个堆的技巧在于判断堆顶和新来的数的大小关系以及两个堆的大小关系。将新数加入堆之后,要保证两个堆的大小之差不超过1.
先判断堆顶数和新数的大小关系,有三种情况:
1、最小堆堆顶小于新数时,说明新数在所有数的后半部分;
2、最小堆堆顶大于新数,但最大堆堆顶小于新数时,说明新数处在最小堆堆顶或者最大堆堆顶,也就是一半的位置;
3、最大堆堆顶大于新数时,说明新数在所有数的前半部分。
再判断两个堆的大小关系:
1、如果新数不在中间,那么目标堆不大于另一个堆时,直接将新数插入目标堆;否则将目标堆的堆顶加入到另一个堆,再把新数加入目标堆。
2、如果新数在中间,那么加到较小的堆中。
这样,每次加进来一个数之后,如果两个堆一样大,则中位数是两个堆堆顶的均值,否则中位数是较大堆的堆顶。
参考:
[1] https://segmentfault.com/a/1190000003709954