239. 滑动窗口最大值
给你一个整数数组 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]
示例 3:
输入:nums = [1,-1], k = 1
输出:[1,-1]
示例 4:
输入:nums = [9,11], k = 2
输出:[11]
示例 5:
输入:nums = [4,-2], k = 2
输出:[4]
提示:
- 1 <= nums.length <= 105
- -104 <= nums[i] <= 104
- 1 <= k <= nums.length
方法一:单调栈
维护一个「单调栈」,有如下特点
- 栈是严格单调递减,最大值即为最先入栈的元素
- 栈内都是当前滑动窗口中的数据
参考示例一,步骤如下:
- 1 入栈
- 3 准备入栈,发现 1 比 3 小弹出 1,3 再入栈
- -1 入栈,满足 k = 3,把栈内最先入栈的 3 加入结果集:[3]
- -3 入栈,把栈内最先入栈的 3 加入结果集:[3, 3]
- 5 准备入栈,发现 -3 比 5 小弹出 -3,发现 -1 比 5 小弹出 - 1,发现 3 比 5 小弹出 3,5 再入栈,把 5 加入结果集:[3, 3, 5]
- 3 入栈,把栈内最先入栈的 5 加入结果集:[3, 3, 5, 5]
- 6 准备入栈,发现 5 比 6 小弹出 5,6 再入栈,把 6 加入结果集:[3, 3, 5, 5, 6]
- 7 准备入栈,发现 6 比 7 小弹出 6,7 再入栈,把 7 加入结果集:[3, 3, 5, 5, 6, 7]
当然,示例一少了一种情况,就是最先入栈的元素是否已经超出滑动窗口,为了判断元素是否超出滑动窗口,栈内应该保存元素的角标。
public int[] maxSlidingWindow(int[] nums, int k) {
Deque<Integer> deque = new LinkedList<>();
int[] ans = new int[nums.length - k + 1];
for (int i = 0; i < nums.length; i++) {
// 维持单调递减性
while (!deque.isEmpty() && nums[i] >= nums[deque.peekLast()]) {
deque.pollLast();
}
deque.offerLast(i);
// 元素是否超过滑动窗口
if (i - deque.peekFirst() >= k) {
deque.pollFirst();
}
if (i + 1 >= k) {
ans[i - k + 1] = nums[deque.peekFirst()];
}
}
return ans;
}
- 时间复杂度:O(n)
- 空间复杂度:O(n)