480. 滑动窗口中位数

题目描述：中位数是有序序列最中间的那个数。如果序列的长度是偶数，则没有最中间的数；此时中位数是最中间的两个数的平均数。
例如：
[2,3,4]，中位数是 3
[2,3]，中位数是 (2 + 3) / 2 = 2.5
给你一个数组 nums，有一个长度为 k 的窗口从最左端滑动到最右端。窗口中有 k 个数，每次窗口向右移动 1 位。你的任务是找出每次窗口移动后得到的新窗口中元素的中位数，并输出由它们组成的数组。
解题思路：用两个优先级队列small和large来存储滑动窗口内的数据，small是一个大顶推存储更小的一半元素，large是小顶堆存储更大的一半元素，如果k是偶数，那么中位数就是两个堆顶元素平均数，如果k是奇数，那么就是small的堆顶元素(这种情况下我们让small比large多存储一个元素)，最后通过延迟删除来实现滑动窗口像右滑动，我们用delayed字典存储要删除元素以及对应的次数，每次右滑动时我们就将左边边界划出去的数字的次数加一，因为中位数只和堆顶元素有关，所以只有当堆顶元素为需要删除的元素时，我们才执行真正的删除操作，代码如下：

class DualHeap():
    def __init__(self, k):
        self.k = k
        self.small = [] #存储一半较小的元素，大顶推
        self.large = [] #存储一半较大的元素，小顶堆
        self.smallsize = 0
        self.largesize = 0
        self.delayed = collections.Counter()
        
    def prune(self, heap):
        while heap:
            num = heap[0]
            if heap is self.small:
                num = -num
            if num in self.delayed:
                self.delayed[num] -= 1
                if self.delayed[num] == 0:
                    self.delayed.pop(num)
                heapq.heappop(heap)
            else:
                break
    def makebalance(self):
        if self.smallsize > self.largesize + 1:
            heapq.heappush(self.large, -self.small[0])
            heapq.heappop(self.small)
            self.smallsize -= 1
            self.largesize += 1
            self.prune(self.small)
        elif self.largesize > self.smallsize:
            heapq.heappush(self.small, -self.large[0])
            heapq.heappop(self.large)
            self.smallsize += 1
            self.largesize -=1
            self.prune(self.large)
    
    def insert(self, num):
        if not self.small or num <= -self.small[0]:
            heapq.heappush(self.small, -num)
            self.smallsize += 1
        else:
            heapq.heappush(self.large, num)
            self.largesize += 1
        self.makebalance()
        
    def erase(self, num):
        self.delayed[num] += 1
        if num <= -self.small[0]:
            self.smallsize -= 1
            if num == -self.small[0]:
                self.prune(self.small)
        else:
            self.largesize -= 1
            if num == self.large[0]:
                self.prune(self.large)
        self.makebalance()
    
    def getMedian(self):
        return -self.small[0] if self.k % 2 else (-self.small[0] + self.large[0]) / 2
class Solution:
    def medianSlidingWindow(self, nums: List[int], k: int) -> List[float]:
        dh = DualHeap(k)
        for num in nums[:k]:
            dh.insert(num)
        ans = [dh.getMedian()]
        
        for i in range(k, len(nums)):
            dh.insert(nums[i])
            dh.erase(nums[i-k])
            ans.append(dh.getMedian())
        return ans