滑动窗口最大值

Leetcode 239. 滑动窗口最大值 - 力扣(LeetCode) (leetcode-cn.com)

暴力解法:

        每次在滑动窗口中去找最大值

        时间复杂度:O((N-K+1)*n)

代码如下:

def maxSlidingWindow(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[int]
        """
        res=[]
        for i in range(len(nums)):
            maxValue=-10001
            for j in range(i,i+k):
                if i+k-1<len(nums):
                    if j<len(nums):
                        maxValue=maxValue if maxValue>nums[j] else nums[j]
                else:
                    return res
            res.append(maxValue)
        return res

但是这个会超出时间限制

联想数据结构--大顶堆

        每次在滑动窗口内建立大顶堆 O(logk)

        每次取出堆顶 O(1)

时间复杂度为O(n·logk)

heapq是小顶堆,python里没有可以直接用的库

只扫描一次的做法:12 | 面试题:返回滑动窗口中的最大值 (geekbang.org)

使用一个双端队列维护滑动窗口,该滑动窗口的0号元素始终保持为该滑动窗口的最大值

即新看到的值比队尾元素大时,队尾元素应该出队。

具体如图:

 

 时间复杂度:O(n)

代码如下:

def maxSlidingWindow(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[int]
        """
        if not nums:
            return []
        window,res=[],[]
        for i in range(len(nums)):
            if i>=k and window[0]<i-k+1:
                window.pop(0)
            while window and nums[window[-1]]<=nums[i]:
                window.pop()
            window.append(i)
            if i-k+1>=0:
                res.append(nums[window[0]])
        return res
                

相关问题:

maxpooling

以上是maxpooling在一维向量下的模拟,面试中很容易问到二维的maxpooling如何写。今天刚好读到一篇题解:[Python] Follow-up: 实现二维max pooling - 滑动窗口最大值 - 力扣(LeetCode) (leetcode-cn.com)

算法与测试用例都很清晰。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值