滑动窗口最大值
给定一个数组 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 总是有效的，1 ≤ k ≤ 输入数组的大小，且输入数组不为空。
进阶：
你能在线性时间复杂度内解决此题吗？
思路
维护一个大小为K的最大堆，依此维护一个大小为K的窗口，每次读入一个新数，都把堆中窗口最左边的数扔掉，再把新数加入堆中，这样堆顶就是这个窗口内最大的值。
注意
-结果数组的大小是nums.length + 1 - k， 赋值时下标也是i + 1 - k
1 import java.util.Collections; 2 import java.util.PriorityQueue; 3 4 public class Solution { 5 public int[] maxSlidingWindow(int[] nums, int k) { 6 if(nums == null || nums.length == 0) return new int[0]; 7 PriorityQueue<Integer> pq = new PriorityQueue<Integer>(Collections.reverseOrder()); 8 int[] res = new int[nums.length + 1 - k]; 9 for(int i = 0; i < nums.length; i++){ 10 // 把窗口最左边的数去掉 11 if(i >= k) pq.remove(nums[i - k]); 12 // 把新的数加入窗口的堆中 13 pq.offer(nums[i]); 14 // 堆顶就是窗口的最大值 15 if(i + 1 >= k) res[i + 1 - k] = pq.peek(); 16 } 17 return res; 18 } 19 }
双向队列
复杂度
时间 O(N) 空间 O(K)
思路
我们用双向队列可以在O(N)时间内解决这题。当我们遇到新的数时，将新的数和双向队列的末尾比较，如果末尾比新数小，则把末尾扔掉，直到该队列的末尾比新数大或者队列为空的时候才住手。这样，我们可以保证队列里的元素是从头到尾降序的，由于队列里只有窗口内的数，所以他们其实就是窗口内第一大，第二大，第三大...的数。保持队列里只有窗口内数的方法和上个解法一样，也是每来一个新的把窗口最左边的扔掉，然后把新的加进去。然而由于我们在加新数的时候，已经把很多没用的数给扔了，这样队列头部的数并不一定是窗口最左边的数。这里的技巧是，我们队列中存的是那个数在原数组中的下标，这样我们既可以直到这个数的值，也可以知道该数是不是窗口最左边的数。这里为什么时间复杂度是O(N)呢？因为每个数只可能被操作最多两次，一次是加入队列的时候，一次是因为有别的更大数在后面，所以被扔掉，或者因为出了窗口而被扔掉。
1 public class Solution { 2 public int[] maxSlidingWindow(int[] nums, int k) { 3 if(nums == null || nums.length == 0) return new int[0]; 4 LinkedList<Integer> deque = new LinkedList<Integer>(); 5 int[] res = new int[nums.length + 1 - k]; 6 for(int i = 0; i < nums.length; i++){ 7 // 每当新数进来时，如果发现队列头部的数的下标，是窗口最左边数的下标，则扔掉 8 if(!deque.isEmpty() && deque.peekFirst() == i - k) deque.poll(); 9 // 把队列尾部所有比新数小的都扔掉，保证队列是降序的 10 while(!deque.isEmpty() && nums[deque.peekLast()] < nums[i]) deque.removeLast(); 11 // 加入新数 12 deque.offerLast(i); 13 // 队列头部就是该窗口内第一大的 14 if((i + 1) >= k) res[i + 1 - k] = nums[deque.peek()]; 15 } 16 return res; 17 } 18 }