题目
给定一个数组 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,用来保存窗口内数据,其中windows中保存的是元素下标
- 依次加入堆,如果size超过k,就把堆内最小的元素,剔除。
- 记录在结果数组中,并返回
public int[] maxSlidingWindowByDequeue(int[] nums, int k) {
ArrayDeque<Integer> windows = new ArrayDeque<>(k);
List<Integer> result = new ArrayList<>();
for (int i = 0; i < nums.length; i++) {
int x = nums[i];
//淘汰掉前面的老元素
if(i >= k && windows.getFirst() <= i - k){
windows.pop();
}
//如果窗口内有小于当前元素的老元素,那么全部剔除, 从后开始比较,并且删除
while(!windows.isEmpty() && nums[windows.getLast()]<= x){
windows.removeLast();
}
//加入队列
windows.add(i);
//当窗口移动到k时才往结果集放元素
if(i>=k-1){
result.add(nums[windows.getFirst()]);
}
}
int a[] = new int[result.size()];
for (int i = 0; i < result.size(); i++) {
a[i] = result.get(i);
}
return a;
}