单调队列
设计单调队列的时候,pop,和push操作要保持如下规则:
pop(value)
:如果窗口移除的元素value等于单调队列的出口元素,那么队列弹出元素,否则不用任何操作
push(value)
:如果push的元素value大于入口元素的数值,那么就将队列入口的元素弹出,直到push元素的数值小于等于队列入口元素的数值为止
保持如上规则,每次窗口移动的时候,只要问que.front()
就可以返回当前窗口的最大值。
// 手动实现一个单调队列
class MyQueue{
Deque<Integer> deque = new LinkedList<>();
void pop(int val) {
if(!deque.isEmpty() && val == deque.getFirst()){
// 返回并删除队列的开头,如果队列为空,则返回null
deque.poll();
}
}
void push(int val) {
while(!deque.isEmpty() && val > deque.getLast()){
deque.removeLast();
}
deque.add(val);
}
// 队顶元素始终为最大值
int peek() {
return deque.getFirst();
}
}
// 自定义数组
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
if(nums.length == 1){
return nums;
}
int len = nums.length - k + 1;
int[] res = new int[len];
MyQueue queue = new MyQueue();
for(int i = 0; i < k; i++){
queue.push(nums[i]);
}
int count = 0;
res[count++] = queue.peek();
for(int i = k; i < nums.length; i++){
queue.pop(nums[i - k]);
queue.push(nums[i]);
res[count++] = queue.peek();
}
return res;
}
}
双端队列
用一个单调队列来存储对应的下标,每当窗口滑动的时候,直接取队列的头部指针对应的值放入结果集即可
单调队列类似 (tail -->) 3 --> 2 --> 1 --> 0 (–> head) (右边为头结点,元素存的是下标)
// 用一个单调队列来存储对应的下标,每当窗口滑动的时候,直接取队列的头部指针对应的值放入结果集即可
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
ArrayDeque<Integer> deque = new ArrayDeque<>();
int len = nums.length;
int[] res = new int[len - k + 1];
int count = 0;
for(int i = 0; i < len; i++){
// 保证队列中一直为滑动窗口k个元素,否则弹出队列头元素下标
while(!deque.isEmpty() && deque.peek() < i - k + 1){
deque.poll();
}
// 当队列尾元素对应数组值小于即将入队的元素值,则弹出队列尾元素下标
while(!deque.isEmpty() && nums[deque.peekLast()] < nums[i]){
deque.pollLast();
}
// 数组i下标入队尾
deque.addLast(i);
// 当下标 >= k - 1时,i每增加一次,都会产生一个滑动窗口最大值
if(i >= k - 1){
res[count++] = nums[deque.peek()];
}
}
return res;
}
}