文章目录
- 1理解题目
- 2 思路
- 2.1暴力求解
- 2.2双端队列
1理解题目
输入：整数数组nums，滑动窗口大小k
输出：整数数组
规则：在一个窗口内只能看到k个数，找一个最大的数，添加到返回数组中。每次滑动向右滑动一步。
Input: nums = [1,3,-1,-3,5,3,6,7], k = 3
Output: [3,3,5,5,6,7]
Explanation:
Window position Max
[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 思路
2.1暴力求解
只能想到暴力的思路。每次窗口范围内比较k个元素大小。所以时间复杂度是O(nk)。
class Solution {public int[] maxSlidingWindow(int[] nums, int k) {if(nums==null || nums.length==0) return null;if(k==0) return nums;int n = nums.length;int[] r = new int[n-k+1];for(int i=0;i<n-k+1;i++){r[i] = nums[i];for(int j=i;j<i+k;j++){r[i] = Math.max(r[i],nums[j]);}}return r;}}
2.2双端队列
看了力扣官方解释想起来了，当时做的时候确实蒙圈。
队列是一个双端队列。队列中存放的是当前窗口内的k个元素的下标。注意这里不存元素，存下标。队列头部的元素是窗口范围内，最大值元素的下标。例如：nums=[1,3,-1,-3,5,3,6,7] ,在{-1,-3,5}这个窗口范围内，队列头部存放的元素值是4。队列尾部存放的元素是当前窗口内最大值元素的下标。最大值是5，下标是4。
每次添加下标为i的元素的时候：
1 检查队列头部元素是不是等于i-k。是，就删除。
2 不断拿nums[i]与队尾元素比较。如果nums[i]>nums[队尾元素]，则删除队尾元素，在队尾插入i。否则不做任何操作。
队列中元素的nums值是不断递减的。
class Solution {private ArrayDeque<Integer> deq = new ArrayDeque<Integer>();private int[] nums;private int k;public int[] maxSlidingWindow(int[] nums, int k) {if(nums==null || nums.length==0) return null;if(k==0) return nums;int n = nums.length;this.nums = nums;this.k = k;int[] r = new int[n-k+1];deq = new ArrayDeque<Integer>();for(int i=0;i<k;i++){add(i);}r[0] = nums[deq.getFirst()];for(int i=k;i<n;i++){add(i);r[i-k+1] = nums[deq.getFirst()];}return r;}private void add(int i){if(!deq.isEmpty() && deq.getFirst()==i-k){deq.removeFirst();}while(!deq.isEmpty() && nums[deq.getLast()]<nums[i]){deq.removeLast();}deq.addLast(i);}
}
)
