题目描述
给你一个整数数组 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
代码及注释
func maxSlidingWindow(nums []int, k int) []int {
if len(nums) == 0 {
return []int{}
}
// 定义双端队列
deque := make([]int, 0)
// 结果数组
res := make([]int, 0)
// 遍历数组
for i := 0; i < len(nums); i++ {
// 移除队列中不在滑动窗口中的元素
if len(deque) > 0 && deque[0] < i-k+1 {
deque = deque[1:]
}
// 移除队列中所有小于当前元素的元素,因为它们不可能是最大值
for len(deque) > 0 && nums[deque[len(deque)-1]] < nums[i] {
deque = deque[:len(deque)-1]
}
// 将当前元素添加到队列中
deque = append(deque, i)
// 如果窗口已经形成,将当前窗口的最大值添加到结果数组中
if i >= k-1 {
res = append(res, nums[deque[0]])
}
}
return res
}
代码解释
-
定义双端队列:
deque := make([]int, 0)
使用双端队列来存储可能的最大值的索引。
-
移除队列中不在滑动窗口中的元素:
if len(deque) > 0 && deque[0] < i-k+1 { deque = deque[1:] }
确保队列中的元素仅包含在当前窗口内的索引。
-
移除队列中所有小于当前元素的元素:
for len(deque) > 0 && nums[deque[len(deque)-1]] < nums[i] { deque = deque[:len(deque)-1] }
这一步保证队列中的元素按降序排列。
-
将当前元素添加到队列中:
deque = append(deque, i)
将当前元素的索引添加到队列的末尾。
-
将当前窗口的最大值添加到结果数组中:
if i >= k-1 { res = append(res, nums[deque[0]]) }
当窗口大小达到k时,将队列的第一个元素(即当前窗口的最大值)添加到结果数组中。
该算法的时间复杂度为O(n),其中n是数组的长度。