滑动窗口问题:
现在有一个数组a,然后每次移动k个元素,记录每次移动时候最大的值,进行返回
例如:数组a[]={1,3,-1,-3,5,3,6,7},每次移动3个元素
滑动窗口的位置 最大值
--------------- -----
[1 3 -1] -3 5 3 6 7 3
1 [3 -1 -3] 5 3 6 7 3
1 3 [-1 -3 5] 3 6 7 5
1 3 -1 [-3 5 3] 6 7 5
1 3 -1 -3 [5 3 6] 7 6
1 3 -1 -3 5 [3 6 7] 7
代码实现可以使用双端队列来实现,此处双端队列里面记录的是数组的下标
public int[] maxSlidingWindow(int[] nums, int k) {
if (nums == null || k < 1 || nums.length < k) {
return new int[0];
}
//定义一个双端队列
Deque<Integer> deque=new ArrayDeque<Integer>();
//定义一个返回的数组
int [] res=new int [nums.length-k+1];
//定义一个index,作为返回的序列
int index=0;
for(int i=0;i<nums.length;i++){
//判断队列的头是否需要出队列,也就是移动了窗口,队列头的元素要出队列
if(!deque.isEmpty()&&deque.peekFirst()<=i-k){
deque.pollFirst();
}
//新元素进来比较队列中尾部的元素,如果比队列尾部元素大则移除
while(!deque.isEmpty()&&nums[deque.peekLast()]<=nums[i]){
deque.pollLast();
}
//元素添加到队列尾部
deque.addLast(i);
//每次移动窗口都记录最大的值
if(i>=k-1){
res[index++]=nums[deque.peekFirst()];
}
}
return res;
}