题目描述
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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