剑指 Offer 41. 数据流中的中位数

剑指 Offer 41. 数据流中的中位数

题目描述

from heapq import *
class MedianFinder:

    def __init__(self):
        """
        initialize your data structure here.
        """
        self.Bigger = [] # 小顶堆，保存较大的一半
        self.Smaller = [] # 大顶堆，保存较小的一半


    def addNum(self, num: int) -> None:
        if len(self.Bigger) == len(self.Smaller):
            heappush(self.Bigger, -heappushpop(self.Smaller, - num))
        else:
            heappush(self.Smaller, -heappushpop(self.Bigger, num))

    def findMedian(self) -> float:
        return self.Bigger[0] if len(self.Bigger) > len(self.Smaller) else (self.Bigger[0]- self.Smaller[0])/2

思路：
维护一个大顶堆存较小数，和一个小顶堆存较大数，堆顶都是最接近另一堆的值，利用heapq实现，维护堆是相差不超过一，大顶堆由存入相反数实现。