一、问题描述:
给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回 滑动窗口中的最大值 。
二、解题思路:
- 利用双端队列的特性,通过动态维护队列的单调递减性质,求解滑动窗口最大值的问题。
- 具体而言,算法的思路如下:
①创建一个双端队列q,用于存储当前窗口内的元素。队列会保持单调递减的特性,以便在队列首部始终保留当前窗口内的最大值。
②创建一个数组max,用于存储所有的最大值,其长度为nums.length - k + 1,其中k为窗口大小。
③遍历数组nums,对于每个元素nums[i]:
(1)将窗口向后滑动一位,将新数据加入队列q。在加入之前,需要维护队列的单调性,即移除队尾小于等于nums[i]的元素。
(2)如果窗口的大小已经达到k,需要判断队列首部的元素是否需要被移除,因为它可能是当前窗口内的最大值。
(3)当遍历到第k个元素时(即窗口大小已达到k),将队列首部的元素作为当前窗口内的最大值存储在max数组中。
④最终返回数组max,其中存储了滑动窗口在不同位置时的最大值。 - 图解分析:
三、代码示例:
//时间复杂度为O(n)
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
Deque<Integer> q = new LinkedList<>();//队列是单调减的,会将当前窗口内的最大值保留下来
int[] max = new int[nums.length - k + 1];//记录所有的最大值,长度为nums.length - k + 1
int j = 0;
for(int i = 0; i < nums.length; i++){
//窗口向后滑动一位,将新数据加入队列,并且要维护队列的单调性,把比nums[i]小的元素都弹出来
while(!q.isEmpty() && nums[i] > q.peekLast()) q.pollLast();
q.offerLast(nums[i]);
//判断队列首元素是否需要被弹出,每次移动窗口都要舍弃最左侧的数,可能就是当前队列中的最大值,如果是应该弹出
if(i >= k && q.peekFirst() == nums[i - k]) q.pollFirst();
//最大值就是队列首元素
if(i >= k - 1) max[j++] = q.peekFirst();
//System.out.println(q.size());
}
return max;
}
}
- 时间复杂度分析:
在代码中,使用了一个while循环来维护双端队列的单调性,其时间复杂度为O(n),其中n为输入数组nums的长度。
除去while循环外,算法只包含一个for循环,其时间复杂度为O(n)。在每个元素上最多进行一次入队和一次出队操作。
因此,整个算法的时间复杂度为O(n) + O(n) = O(n)。
四、补充与总结:
熟悉队列的特点和用法。