leetcode 295.Find Median from Data Stream

优先队列

题意:

实现以下两种操作

1.插入一个数到集合里(可重复)

2.询问集合里的中位数

思路:

优先队列的经典题目.

建立两个堆(优先队列),一个大根堆,一个小根堆,它们各储存集合里一半的数.

小根堆储存大的那半,大根堆储存小的那半.

显然这时候两个堆的堆顶就是最中间的两个数,中位数就是它们的和的一半,或者它们其中一个.

插入的时候,先判断插入到大根堆还是小根堆,再调整两个堆的size,使得它们的size最多相差1.

调整的方法是把小根堆堆顶push到大根堆,或者大根堆堆顶push到小根堆.

询问操作是O(1),插入操作最多push两次,因此是O(logn).

总结:

维护两个堆,一个大根堆记录集合小的那半,一个小根堆记录大的那半,中位数从堆顶得到.