1.题目描述
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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]}
2.算法描述
用
双
端
队
列
,
队
列
的
队
头
是
滑
动
窗
口
最
大
值
的
下
标
。
\red{双端队列,队列的队头是滑动窗口最大值的下标。}
双端队列,队列的队头是滑动窗口最大值的下标。
1.
当
窗
口
滑
动
一
次
,
把
队
列
中
比
新
元
素
小
的
元
素
的
下
标
弹
出
,
保
证
了
队
列
队
头
是
窗
口
元
素
值
最
大
的
下
标
。
\red{1.当窗口滑动一次,把队列中比新元素小的元素的下标弹出,保证了队列队头是窗口元素值最大的下标。}
1.当窗口滑动一次,把队列中比新元素小的元素的下标弹出,保证了队列队头是窗口元素值最大的下标。
2.
当
窗
口
滑
动
一
次
,
把
过
期
的
队
头
元
素
弹
出
。
\red{2.当窗口滑动一次,把过期的队头元素弹出。}
2.当窗口滑动一次,把过期的队头元素弹出。
3.代码描述
3.1.Java代码
import java.util.*;
public class Solution {
public ArrayList<Integer> maxInWindows(int [] num, int size){
ArrayList<Integer> ans = new ArrayList<>();
if(num==null || size==0 || size>num.length)//此类情况 返回空的列表
return ans;
LinkedList<Integer> dq = new LinkedList<>();//双端队列 装的是元素下标
for(int i=0; i<num.length; i++){//扫描一遍元素
//当窗口要将新的元素包含进来的时候,把窗口中比新元素值小的元素的下标都弹出
//保证队列的队头是窗口元素值最大的下标
while(!dq.isEmpty() && num[i]>=num[dq.peekLast()])
dq.pollLast();
dq.addLast(i);//新元素下标进入队尾
//如果队列包含的元素个数大于size 说明当前窗口左端已经把包括队头 弹出
if(i-dq.peekFirst()+1>size)
dq.pollFirst();
//当窗口中有size个元素的时候,此时队头的是对应窗口最大值的下标
if(i+1>=size)
ans.add(num[dq.peekFirst()]);
}
return ans;
}
}
3.2.Python代码
# -*- coding:utf-8 -*-
class Solution:
def maxInWindows(self, num, size):
ans = []
if not num or size==0 or size>len(num):
return ans
dq = []
for i in range(len(num)):
while dq and num[i]>=num[dq[-1]]:
dq.pop()
dq.append(i)
if i-dq[0]+1>size:
dq.pop(0)
if i+1>=size:
ans.append(num[dq[0]])
return ans