480. Sliding Window Median

ujn20161222

于 2024-10-25 10:07:01 发布

阅读量195

点赞数 2

CC 4.0 BY-SA版权

文章标签： leetcode

本文链接：https://blog.csdn.net/ujn20161222/article/details/143227346

480. Sliding Window Median

from sortedcontainers import SortedList
class Solution:
    def medianSlidingWindow(self, nums: List[int], k: int) -> List[float]:
        n = len(nums)
        acc= SortedList(nums[:k-1])
        res=[]
        for i in range(k-1,len(nums)):
            acc.add(nums[i])
            res.append((acc[k//2]+acc[(k-1)//2])/2)
            acc.discard(nums[i-k+1])
        return res