python leetcode 480. Sliding Window Medi

最新推荐文章于 2022-05-17 17:24:33 发布

原创最新推荐文章于 2022-05-17 17:24:33 发布 · 665 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #leetcode

python 同时被 2 个专栏收录

263 篇文章

订阅专栏

leetcode

221 篇文章

订阅专栏

本文介绍了一种使用bisect模块在滑动窗口中高效查找中位数的方法。通过维护一个有序数组，每次窗口滑动时更新数组，实现O(log k)的时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

用bisect维护即可

class Solution:
    def medianSlidingWindow(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[float]
        """
        import bisect
        n=len(nums)
        res=[]
        tmp=sorted(nums[:k])
        res.append(self.get(tmp)*1.0)
        for i in range(n-k):
            index=bisect.bisect_left(tmp,nums[i])
            del tmp[index]
            bisect.insort(tmp,nums[i+k])
            res.append(self.get(tmp)*1.0)
        return res
    def get(self,tmp):
        l=len(tmp)
        if l&1==1: return tmp[l//2]
        else: return tmp[l//2-1]*0.5+tmp[l//2]*0.5