滑动窗口最大值
题目描述:
给你一个整数数组 nums
,有一个大小为 k
的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k
个数字。滑动窗口每次只向右移动一位。最后,返回 滑动窗口中的最大值 。
示例 1:
输入:nums = [1,3,-1,-3,5,3,6,7], k = 3 输出:[3,3,5,5,6,7] 解释: 滑动窗口的位置 最大值 --------------- ----- [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
示例 2:
输入:nums = [1], k = 1 输出:[1]
提示:
1 <= nums.length <= 105
-104 <= nums[i] <= 104
1 <= k <= nums.length
思路分析:
为了解决这个问题,我们可以使用双端队列(deque)来存储滑动窗口中的元素索引,并且保证队列的头部始终是滑动窗口中的最大元素的索引。这样,我们可以在O(1)的时间复杂度内获取到滑动窗口的最大值。
下面是具体的实现步骤:
-
初始化一个双端队列
deque
用于存储元素的索引,以及一个结果数组result
用于存储每个滑动窗口的最大值。 -
遍历数组
nums
,对于每个索引i
:- 如果队列不为空且队列头部索引对应的元素已经不在当前窗口内(即
deque.peek() <= i - k
),则从队列头部移除该索引。 - 从队列尾部开始,移除所有小于当前元素
nums[i]
的索引对应的元素。这是因为这些较小的元素在后续窗口中也不可能是最大值,所以我们可以提前将它们移除,以保持队列的单调性(即队列从头部到尾部对应的元素值是递减的)。 - 将当前元素的索引
i
添加到队列的尾部。 - 如果当前窗口已经完全形成(即
i >= k - 1
),则将队列头部的索引对应的元素值(即当前窗口的最大值)添加到结果数组result
中。
- 如果队列不为空且队列头部索引对应的元素已经不在当前窗口内(即
3. 随着窗口的滑动,我们不断地更新队列,并确保队列的头部始终指向当前窗口的最大值。当窗口大小达到 k
时,我们将队列头部的索引对应的元素值添加到结果数组中。最后,当遍历完整个数组后,我们得到了包含每个滑动窗口最大值的数组,并将其返回。
思路图如下:
代码实现注解:
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
ArrayDeque<Integer> deque = new ArrayDeque<>();
int n = nums.length;
int[] res = new int[n - k + 1];
int a = 0;
for (int i = 0; i < n; i++) {
// 移除窗口之外的元素
if (!deque.isEmpty() && deque.peek() <= i - k) {
deque.poll();
}
// 移除队列中比当前元素小的元素
while (!deque.isEmpty() && nums[deque.peekLast()] < nums[i]) {
deque.pollLast();
}
// 将当前元素加入队列尾部
deque.offer(i);
// 窗口形成后开始填充结果数组
if (i >= k - 1) {
res[a++] = nums[deque.peek()];
}
}
return res;
}
}