优先队列
题意:
实现以下两种操作
1.插入一个数到集合里(可重复)
2.询问集合里的中位数
思路:
优先队列的经典题目.
建立两个堆(优先队列),一个大根堆,一个小根堆,它们各储存集合里一半的数.
小根堆储存大的那半,大根堆储存小的那半.
显然这时候两个堆的堆顶就是最中间的两个数,中位数就是它们的和的一半,或者它们其中一个.
插入的时候,先判断插入到大根堆还是小根堆,再调整两个堆的size,使得它们的size最多相差1.
调整的方法是把小根堆堆顶push到大根堆,或者大根堆堆顶push到小根堆.
询问操作是O(1),插入操作最多push两次,因此是O(logn).
总结:
维护两个堆,一个大根堆记录集合小的那半,一个小根堆记录大的那半,中位数从堆顶得到.