LC 剑指 Offer 59 - I. 滑动窗口的最大值
给定一个数组 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
题解
使用单调栈实现了随意入栈、出栈情况下的 O(1)时间获取 “栈内最大值”
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
if (nums == null || nums.length == 0) {
return new int[0];
}
int[] res = new int[nums.length - k + 1];
LinkedList<Integer> queue = new LinkedList<>();
// 形成滑窗前
for (int i = 0; i < k; i++) {
while (!queue.isEmpty() && nums[i] > queue.getLast()) {
queue.removeLast();
}
queue.addLast(nums[i]);
}
res[0] = queue.getFirst();
// 形成滑窗后
for(int i = k; i < nums.length; i++) {
// 删除元素
if (queue.getFirst().equals(nums[i-k])) {
queue.removeFirst();
}
// 添加元素
while (!queue.isEmpty() && nums[i] > queue.getLast()) {
queue.removeLast();
}
queue.addLast(nums[i]);
res[i-k+1] = queue.getFirst();
}
return res;
}
}