剑指 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
提示:
你可以假设 k 总是有效的,在输入数组不为空的情况下,1 ≤ k ≤ 输入数组的大小。
分析
1. 暴力循环
窗口大小是k,数组长度是n,时间复杂度就是O(KN),空间复杂度O(1)。
2. 单调双端队列
维护一个单调双端队列,如果窗口移动后的值大于队列中的值,就将队列中小于这个数的数字全部从尾部出队。
如果窗口右侧的值小于队列的尾部,直接入队。
如果窗口滑过去了队首元素,就将队首出队。
时间复杂度:每个元素最多入队一次出队一次,所以时间复杂度为O(2N)=O(N).
空间复杂度:维护最大程度为K的队列,空间复杂度为O(K)。
代码
暴力循环
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
if(nums.length == 0) return new int[0];
int[] result = new int[nums.length-k+1];
for(int i = 0; i<result.length; i++){
result[i] = calcMax(nums,i,i+k-1);
}
return result;
}
public int calcMax(int[] nums, int i, int j){
int max = nums[i];
for(; i<=j; i++){
max = Math.max(max,nums[i]);
}
return max;
}
}
单调双端队列
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
//排除特殊情况
if(nums.length == 0) return new int[0];
//初始化
Deque<Integer> d = new LinkedList<>();
int[] result = new int[nums.length-k+1];
for(int i = 0; i<nums.length; i++){
//i是窗口右侧索引。
//如果队列不为空,并且队首索引被滑过去了,就将队首弹出。
if(!d.isEmpty() && d.peekFirst() == i-k) d.pollFirst();
//将所有比当前数值小的数字出队。
while(!d.isEmpty() && nums[i]>=nums[d.peekLast()]) d.pollLast();
//入队
d.offerLast(i);
//检查是否是窗口,如果形成了窗口,就开始保存结果
if(i-k+1>=0){
result[i-k+1] = nums[d.peekFirst()];
}
}
return result;
}
}