Sliding Window Maximum
给定一个数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口 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 ≤ 输入数组的大小,且输入数组不为空。
进阶:
你能在线性时间复杂度内解决此题吗?
分析
双向队列,始终保证降序排列:
- 新元素和队尾元素比较,前者大,则后者出队,前者继续和队尾元素比,直到队列为空或新元素比队尾元素小,将新元素入队。
代码
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
if(nums.length == 0){
return new int[0];
}
Deque<Integer> win = new LinkedList<>();
int[] res = new int[nums.length + 1-k];
for(int i=0;i<nums.length;i++){
if(!win.isEmpty()&&win.peekFirst()==i-k)
win.poll();
while(!win.isEmpty()&&nums[win.peekLast()]<nums[i]){
win.removeLast();
}
win.offerLast(i);
if(i+1>=k){
res[i+1-k]=nums[win.peek()];
}
}
return res;
}
}