239.滑动窗口最大值
给你一个整数数组 nums
,有一个大小为 k
的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k
个数字。滑动窗口每次只向右移动一位。
返回 滑动窗口中的最大值 。
示例 1:
输入: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
示例 2:
输入:nums = [1], k = 1 输出:[1]
提示:
1 <= nums.length <= 105
-104 <= nums[i] <= 104
1 <= k <= nums.length
思路(没有思路)
刚开始想着一点一点移过去,然后暴力算最大值,但是时间复杂度太大,而且只用数组就解决了。
然后想着可以记录最大点的位置,然后等这个点出去了就再找个新的,但是也不靠谱,最后还是要暴力排序。
还有一个想法是把从大到小排列起来,大的走了就删掉,但是又不能直接对数组进行排序,不然滑动窗口会乱。
看了下题解的想法,感觉和排最大的数字有点像,用了一个双端队列来记录可能是最大值的数字,最大的值在队首,后面进来的值如果大于队尾值则将队尾值出队,直到它小于队尾值再入队。
然后如果滑动窗口第一个数和队首元素相同,那么就将队首元素出队再滑动窗口。
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
Deque<Integer> deque = new ArrayDeque<>();
int[] ans = new int[nums.length - k + 1];
for (int i = 0; i < k; i++) {
while (!deque.isEmpty() &&nums[i] > deque.peekLast()) {
deque.pollLast();
}
deque.addLast(nums[i]);
}
ans[0] = deque.peekFirst();
for (int i = k; i < nums.length; i++) {
if (nums[i - k] == deque.peek()) {
deque.pollFirst();
}
while (!deque.isEmpty() &&nums[i] > deque.peekLast()) {
deque.pollLast();
}
deque.addLast(nums[i]);
ans[i - k + 1] = deque.peekFirst();
}
return ans;
}
}
总结
这道题是第一道见过的困难题,感觉这个代表滑动窗口不维护所有的元素这种思路很巧妙。同时,ArrayDeque的一些pop、push、add方法不是很熟练,所以最后索性都用了带有first和last的方法。