Leetcode 239题目:
输入: 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
核心思想是维持deque/双端队列中的元素保持递增or递减。
使用该数据结构的优点是deque在队列两端都可以添加、删除元素,这里借助了它其中4种常数时间复杂度的操作(java):offerLast(n)、getFirst()、pollFirst()、pollLast()。
单调栈介绍详见:https://blog.csdn.net/LutherK/article/details/107023543
另外:单调队列还被应用于“多重背包问题”的优化解法。
在一堆数字中,已知最值,如果给这堆数添加一个数,那么比较一下就可以很快算出最值;但如果减少一个数,就不一定能很快得到最值了,而要遍历所有数重新找最值。
首先考虑,要在O(N)的时间内解决此问题,我们需要3个O(1)的操作帮助我们完成,分别是:
push(n):在队列末尾添加元素n;
max:取当前队列/窗口中的最大值;
pop(n):r若队首元素是n,则删除n;
在Java中的实现如下:
// 用于"单调队列"的成员内部类 private static class MonotonicQueue { Deque<Integer> deque = new ArrayDeque<>(); void push(int n){ while(!deque.isEmpty() && deque.getLast()<n) deque.pollLast(); deque.offerLast(n); } // int max(){ return deque.getFirst(); } // void pop(int n){ if(!deque.isEmpty() && deque.getFirst()==n) deque.pollFirst(); } }4. 完整Java代码
// (2) 双端队列(单调队列) public static int[] maxSlidingWindow2(int[] nums, int k) { int n = nums.length; MonotonicQueue window = new MonotonicQueue(); int[] res = new int[n-k+1]; for(int i=0;i<n;i++){ if(i<k-1) window.push(nums[i]); else { window.push(nums[i]); res[i-k+1] = window.max(); window.pop(nums[i-k+1]); } } return res; } // 用于"单调队列"的成员内部类 private static class MonotonicQueue { Deque<Integer> deque = new ArrayDeque<>(); void push(int n){ while(!deque.isEmpty() && deque.getLast()<n) deque.pollLast(); deque.offerLast(n); } // int max(){ return deque.getFirst(); } // void pop(int n){ if(!deque.isEmpty() && deque.getFirst()==n) deque.pollFirst(); } }复杂度
时间 O(N) 空间 O(K)
思路
我们用双向队列可以在O(N)时间内解决这题。当我们遇到新的数时,将新的数和双向队列的末尾比较,如果末尾比新数小,则把末尾扔掉,直到该队列的末尾比新数大或者队列为空的时候才住手。这样,我们可以保证队列里的元素是从头到尾降序的,由于队列里只有窗口内的数,所以他们其实就是窗口内第一大,第二大,第三大...的数。保持队列里只有窗口内数的方法和上个解法一样,也是每来一个新的把窗口最左边的扔掉,然后把新的加进去。然而由于我们在加新数的时候,已经把很多没用的数给扔了,这样队列头部的数并不一定是窗口最左边的数。这里的技巧是,我们队列中存的是那个数在原数组中的下标,这样我们既可以直到这个数的值,也可以知道该数是不是窗口最左边的数。这里为什么时间复杂度是O(N)呢?因为每个数只可能被操作最多两次,一次是加入队列的时候,一次是因为有别的更大数在后面,所以被扔掉,或者因为出了窗口而被扔掉。
public class Solution { public int[] maxSlidingWindow(int[] nums, int k) { if(nums == null || nums.length == 0) return new int[0]; LinkedList<Integer> deque = new LinkedList<Integer>(); int[] res = new int[nums.length + 1 - k]; for(int i = 0; i < nums.length; i++){ // 每当新数进来时,如果发现队列头部的数的下标,是窗口最左边数的下标,则扔掉 if(!deque.isEmpty() && deque.peekFirst() == i - k) deque.poll(); // 把队列尾部所有比新数小的都扔掉,保证队列是降序的 while(!deque.isEmpty() && nums[deque.peekLast()] < nums[i]) deque.removeLast(); // 加入新数 deque.offerLast(i); // 队列头部就是该窗口内第一大的 if((i + 1) >= k) res[i + 1 - k] = nums[deque.peek()]; } return res; } }
此题还有另外一种解法,时间复杂度也是O(N),可以说是DP,也可是看做“同步双向遍历”。
代码:
// (1)动态规划 // left[i] 表示从窗口开始到下标i的最大值 // right[j] 表示从窗口末尾到索引j的最大值 public static int[] maxSlidingWindow(int[] nums, int k) { int n = nums.length; if (n * k == 0) return new int[0]; if (k == 1) return nums; int[] left = new int[n]; left[0] = nums[0]; int[] right = new int[n]; for (int i = 1; i < n; i++) { if (i%k==0) left[i] = nums[i]; else left[i] = Math.max(left[i-1],nums[i]); int j = n-1-i; if((j+1)%k==0) right[j] = nums[j]; else right[j] = Math.max(right[j+1],nums[j]); } int[] res = new int[n-k+1]; for (int i = 0; i <= n - k; i++) { res[i] = Math.max(right[i],left[i+k-1]); } return res; }