题目描述
给定一个数组 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
题目分析
这个题目主要是要应用双向链表的思想。
为了满足题目要求的线性时间复杂度,我们通过利用双向链表设计一种遍历规则。
首先定义一个双向链表(滑动窗口记录数组的下标),模拟滑动窗口,窗口内的元素从左到右的大小关系为大->小,数据会从窗口的右边进入。
当一个元素来临,如果该元素大于或等于滑动窗口的最右元素,则将最右元素弹出,依次直到最右元素大于该元素或滑动窗口为空,然后将该元素压入滑动窗口。
然后,检查最左边元素是否符合要求(是否过期),如果过期,则将其弹出
将当前窗口(已形成窗口)的最左元素(最大元素)添加到结果数组中。
代码
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
if(nums.length == 0) return nums;
LinkedList<Integer> window = new LinkedList<Integer>();
//记录数组中含有多少个滑动窗口
int[] res = new int[nums.length - k + 1];
int index = 0;
for(int i = 0;i < nums.length;i++){
//弹出不符合规则的元素
//链表不为空并且数组中的值大于链表头部的值 就弹出链表尾部的元素
while(!window.isEmpty() &&
nums[i] >= nums[window.peekLast()]){
window.pollLast();
}
window.addLast(i);//否则将比链表头部值小的值 从链表尾压入
//清除过期元素,链表的长度不能超过滑动窗口值
if(i - window.peekFirst() == k){
window.pollFirst();
}
//将链表头存为当时滑动窗口的最大值
if(i >= k - 1){
res[index++] = nums[window.peekFirst()];
}
}
return res;
}
}