给你一个整数数组 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sliding-window-maximum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
本题在 剑指 Offer 里面有做过,大致思路记住了,但是编码的时候还是遇到了一些些的问题,所以再写一遍博客把,加深一下印象,本题使用双端队列来维护窗口中的最大值,当插入的数字与队列尾端相比大的时候,让队列出队,知道遇到比该数字大的队列元素,若为空,直接入队。
本题分为两个阶段,第一个阶段,滑动窗口未形成,当 i 小于 k 的时候,窗口未形成,进入准备,每次入队一个元素,找到首窗口的最大值,然后定义一个数组,长度为 length - k + 1,用来记录每个窗口的最大值,每次遍历,首先判断当前队首元素是不是窗口的头的位置,如果是,让窗口头的元素出队,否则,还继续上面的步骤,每次记录最大的元素也就是队首元素,放入数组。
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
Deque<Integer> deque = new LinkedList<>();
for (int i = 0; i < k; i++) {
while (!deque.isEmpty() && deque.peekLast() < nums[i]) {
deque.pollLast();
}
deque.offerLast(nums[i]);
}
int[] res = new int[nums.length - k + 1];
res[0] = deque.peekFirst();
for (int i = k; i < nums.length; i++) {
if (nums[i - k] == deque.peekFirst()) {
deque.pollFirst();
}
while (!deque.isEmpty() && deque.peekLast() < nums[i]) {
deque.pollLast();
}
deque.offerLast(nums[i]);
res[i - k + 1] = deque.peekFirst();
}
return res;
}
}