描述
给定一个数组 nums 和滑动窗口的大小 k,请找出所有滑动窗口里的最大值。
示例:
输入: 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
提示:
- 你可以假设 k 总是有效的,在输入数组不为空的情况下,1 ≤ k ≤ 输入数组的大小。
双端队列解法
思路:
i,j分别代表滑动窗口的左右边界,初始化i=1-k,j=0,当还未形成窗口时,将遍历到的元素加入到双端队列deque,并不断
删除队列中比nums[j]小的数,保证队列的单调递减,将nums[j]加入到队列,此时队列的队头元素即为滑动窗口最大值
//双端队列解法
public int[] maxSlidingWindow(int[] nums, int k) {
if (nums.length == 0 || k == 0) return new int[0];
Deque<Integer> deque = new LinkedList<>();
int[] res = new int[nums.length - k + 1];
for (int i = 1 - k, j = 0; j < nums.length; i++, j++) {
//删除deque中的nums[i-1](形成窗口后)
if (i > 0 && deque.peekFirst() == nums[i - 1])
deque.removeFirst();
//删除deque中比nums[j]小的数,保证deque递减(形成窗口前和形成窗口后都需要)
while (!deque.isEmpty() && deque.peekLast() < nums[j])
deque.removeLast();
deque.add(nums[j]); //将新的元素加入deque
if (i >= 0)
res[i]=deque.peekFirst(); //记录最大值
}
return res;
}
可以将 “未形成窗口” 和 “形成窗口后” 两个阶段拆分到两个循环里实现。代码虽变长,但减少了冗余的判断操作。
public int[] maxSlidingWindow(int[] nums, int k) {
if (nums.length == 0 || k == 0) return new int[0];
Deque<Integer> deque = new LinkedList<>();
int[] res = new int[nums.length - k + 1];
//此时i为右边界
//未形成窗口前
for (int i = 0; i < k; i++) {
while (!deque.isEmpty() && deque.peekLast() < nums[i])
deque.removeLast();
deque.add(nums[i]);
}
res[0] = deque.peekFirst();
//形成窗口后
for (int i = k; i < nums.length; i++) {
if (deque.peekFirst() == nums[i - k])
deque.removeFirst();
while (!deque.isEmpty() && deque.peekLast() < nums[i])
deque.removeLast();
deque.add(nums[i]);
res[i - k + 1] = deque.peekFirst();
}
return res;
}
暴力解法
思路:
编写一个max函数用于计算数组指定范围的最大值,然后在maxSlidingWindow函数中不断遍历数组计算区间最大值
class Solution {
//求数组指定范围的最大值
public int max(int arr[], int i, int j) {
int max = Integer.MIN_VALUE;
for (int k = i; k <= j; k++) {
if (arr[k] > max) {
max = arr[k];
}
}
return max;
}
public int[] maxSlidingWindow(int[] nums, int k) {
if (nums.length == 0) return new int[0];
int[] res = new int[nums.length - k + 1];
for (int i = 0; i < nums.length - k + 1; i++) {
res[i] = max(nums, i, i + k - 1);
}
return res;
}
}