题目
给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回 滑动窗口中的最大值 。
示例 1:
输入: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
示例 2:
输入:nums = [1], k = 1
输出:[1]
来源:力扣(LeetCode)
链接:原题链接
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
首先,单调队列是从大到小或从小到大的队列。本题滑动窗口每次右移一位,保证当前窗口大小固定,这样的话,如果每次移动都比较出当前窗口的最大值,重复比较的次数太多了。
而单调队列,将元素从大到小的进入单调队列,将要入队的元素如果大于队列当前末尾元素,就删除队列末尾元素,只要将要入队的元素小于等于队列末尾元素,就入队,每次将队列变化后的队首元素存进一个list,list里的值就是滑动窗口每次移动之后的最大值。
下面代码中已经写了详细的注释
代码
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
MyQue que = new MyQue();
List<Integer> list = new ArrayList<>();
for(int i = 0; i < k; i++) {
//将前k个元素按单调队列方式入队
que.push(nums[i]);
}
list.add(que.front());//将前k个元素的最大值存进list集合中
for(int i = k; i < nums.length; i++) {
//之后每次循环,都在对头删除,在队尾入队
//但是是否删除和如何入队,按照我们写的类中方法的条件进行
//每次又移一次后,都将变化后的队列中最大值存进list集合
que.pop(nums[i - k]);
que.push(nums[i]);
list.add(que.front());
}
int[] arr = new int[list.size()];
for(int i = 0; i < arr.length; i++) {
arr[i] = list.get(i);
}
return arr;
}
}
//自己创建一个操作类,写出我们需要的入队出队方式
//底层用Deque实现,双端队列可以在队列两端都进行插入删除操作
class MyQue {
Deque<Integer> que = new LinkedList();
public void pop(int value) {
if(!que.isEmpty() && value == que.getFirst()) {
//数组中元素等于队列中第一个元素,就出队第一个元素
//队列第一个元素是队列中最大的元素
que.removeFirst();
}
}
public void push(int value) {
while(!que.isEmpty() && value > que.getLast()) {
//只要数据当前元素大于队列末尾元素,就删除队列末尾元素
que.removeLast();
}
//元素不大于队列末尾后,就将元素添加到队列末尾
que.addLast(value);
}
public int front() {
//返回队列的第一个元素(第一个就是队列的最大值)
return que.getFirst();
}
}