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]
提示:
1 <= nums.length <= 105
-104 <= nums[i] <= 104
1 <= k <= nums.length
解题思路与代码实现
使用单调队列来记录当前窗口内滑动窗口的最大值的下标,以方便查找
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
// 辅助双端队列实现单调队列,记录窗口内非递增序列的下标,这样队头元素对应的下标即窗口内最大元素下标
LinkedList<Integer> list = new LinkedList<>();
int[] res = new int[nums.length - k + 1]; // 保存最终结果
for (int i = 0; i < nums.length; i++) {
// 队列不为空,检查队头元素(即窗口最大元素下标)是否已过期
if (!list.isEmpty()) {
Integer first = list.getFirst(); // 获取队头元素
if (first < i - k + 1) { // 窗口区间为[i-k+1, i]
list.removeFirst(); // 移除过期的队头元素
}
}
// 在队尾插入新元素,并维持非递增序列
// 检查是否需要移除部分队尾元素以维持非递增序列
while (!list.isEmpty()) {
Integer last = list.getLast(); // 获取队尾元素
if (nums[i] > nums[last]) { // 要插入的元素大于队尾元素
list.removeLast(); // 移除队尾元素
} else {
break;
}
}
list.addLast(i); // 队尾插入新元素
// 记录当前窗口元素最大值
if (i >= k - 1) {
// 记录当前窗口元素最大值
res[i - (k - 1)] = nums[list.getFirst()];
}
}
return res;
}
}
踩坑点
使用单调队列来记录当前窗口内滑动窗口的最大值的下标,而非记录当前窗口内滑动窗口的最大值