题目
思路
前面出,后面进。这已经不能再明显了,我们显然要使用队列这种数据结构了。
难点是如何求一个区间里的最大值呢?
此时我们需要一个队列,这个队列呢,放进去窗口里的元素,然后随着窗口的移动,队列也一进一出,每次移动之后,队列告诉我们里面的最大值是什么。
其实队列没有必要维护窗口里的所有元素,只需要维护有可能成为窗口里最大值的元素就可以了,同时保证队里里的元素数值是由大到小的。
那么这个维护元素单调递减的队列就叫做单调队列–头部为最大的,尾部放最小的。
队列代码实现
class MyQueue{
Deque<Integer> deque = new LinkedList<>();
void poll(int val){
if(!deque.isEmpty() && val==deque.peek()){
deque.poll();
}
}
void add(int val){
while(!deque.isEmpty() && val>deque.getLast()){
deque.removeLast();
}
deque.add(val);
}
int peek() {
return deque.peek();
}
}
用这个队列完成操作的部分比较简单,看着写代码即可。只需要注意是先删再加,不然最大值不是我们想要的。
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
if (nums.length == 1) {
return nums;
}
//从k到结尾放的元素长度
int len = nums.length - k + 1;
//存放结果元素的数组
int[] res = new int[len];
int num = 0;
//自定义队列
MyQueue myQueue = new MyQueue();
//先放入k个值
for(int i=0; i<k; i++){
myQueue.add(nums[i]);
}
//将最大值放入
res[num++] = myQueue.peek();
//继续获得最大值
for(int i=k; i<nums.length; i++){
//滑动窗口移除最前面的元素
myQueue.poll(nums[i - k]);
//滑动窗口加入最后面的元素
myQueue.add(nums[i]);
//记录对应的最大值
res[num++] = myQueue.peek();
}
return res;
}
}
附录-对应方法
offer()方法在可能的情况下插入一个元素,否则返回false,这和Collection.add方法不同,后者只能通过抛出一个uncheck Exception来表示插入元素失败(什么是不检查异常?下次总结),offer()方法是为了正常情况下的“失败”而设计的,而不是异常情况,比如固定容量的队列。
remove()和poll()方法删除并且返回队首元素,至于从队列中删除哪一个元素就和队列排序策略的函数实现有关了(有各种实现),remove()和poll()的不同仅仅在于,当队列为空时他们的行为不同:remove()抛出异常,poll()返回null。
element()和peek()方法返回但不删除队首元素。
getLast()方法用于检索双端队列的最后一个元素,但不会从此双端队列中删除该元素。
add()方法用于在双端队列的末尾添加给定元素。
removeLast()方法用于返回最后一个元素,但要从此双端队列移除该元素。
参考:代码随想录