剑指offer:滑动窗口的最大值

题目描述
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小3,那么一共存在6个滑动窗口,他们的最大值分别为{4,4,6,6,6,5}; 针对数组{2,3,4,2,6,2,5,1}的滑动窗口有以下6个: {[2,3,4],2,6,2,5,1}, {2,[3,4,2],6,2,5,1}, {2,3,[4,2,6],2,5,1}, {2,3,4,[2,6,2],5,1}, {2,3,4,2,[6,2,5],1}, {2,3,4,2,6,[2,5,1]}。

方法一
遍历数组依次计算最大值

# -*- coding:utf-8 -*-
class Solution:
    def maxInWindows(self, num, size):
        # write code here
        i=0
        res=[]
        if size==0:
            return []
        while i+size-1<len(num):
            res.append(max(num[i:i+size]))
            i+=1
        return res

方法二
优化以上方法,使用index标记当前遍历到的最大元素,如果不在窗口内就求出新窗口内最大值作为新的最大值,否则将其与当前窗口最后一个值比较大小。

# -*- coding:utf-8 -*-
class Solution:
    def maxInWindows(self, num, size):
        # write code here
        i=0
        index=-1
        res=[]
        if size==0:
            return []
        #index标记当前遍历的最大值,如果超过窗口范围或出现新的更大值则修改index
        while i<=len(num)-size:
            #如果index已经超出窗口范围
            if index<i:
                max=num[i]
                for j in range(i,i+size):
                    if num[j]>max:
                        max=num[j]
                        index=j
            else:#index未超出窗口范围,则比较当前窗口最后一个值是否大于index所指值
                if num[i+size-1]>max:
                    max=num[i+size-1]
                    index=i+size-1
            res.append(max)
            i+=1
        return res

方法三
双端队列,每遍历一个元素则将其下标加入队列中,
从队列末尾开始,删除比当前元素小的元素,因为这些元素不可能成为窗口内的最大值,保证队首元素是之前遍历到的最大值;
从队首开始,如果队首指向的元素和当前元素距离超过窗口大小,则删除队首元素,第二个元素成为之前遍历的最大元素;
参考:
牛客网讨论

# -*- coding:utf-8 -*-
class Solution:
    def maxInWindows(self, num, size):
        # write code here
        i=0
        res=[]
        if size==0:
            return []
        #双端队列
        queue=[]
        while i<len(num):
            while queue and num[queue[-1]]<=num[i]:
                #队列不为空且队列最后的数值小于当前元素时从队列中删除最后队尾元素,保证队列中指向的元素都大于当前元素
                queue.pop()
            while queue and i-queue[0]+1>size:
                #队列不为空且当前元素和队首元素之间距离超过一个窗口大小,即最大元素已经失效
                queue.pop(0)
            queue.append(i)
            if i+1>=size:
                res.append(num[queue[0]])
            i+=1
        return res
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值