题目描述
给定一个数组 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 ≤ 输入数组的大小,且输入数组不为空。
进阶:
你能在线性时间复杂度内解决此题吗?
分析解答
这是一道标记为困难
的题目,乍一看会发现有一些坑:线性复杂度以及数组内容的动态增减。由于这道题对应的标签是堆
,所以我们先考虑用堆来解决。维护一个最大堆,每次往右移动一格时把最新的值入堆,然后取出最大值并判断该值是否在我们当前的窗口内。我们可以引入一个额外的计数器作为垃圾回收站,每次往右移动后最左边的值就离开窗口,在计数器中将该值的计数加一,表明回收站中又多了一个该值。然后在取出最大值时判断下回收站中的计数是否为0,不为0的话表明当前这个最大值应该是已经丢弃过的,无法使用,应该继续取下一个最大值,直到对应值在回收站中的计数不为0。这种做法只需一次遍历,但需要维护一个大小为**k*的堆,每次进堆和出堆的平均时间是logk
,因此总耗时实际上是O(Nlogk)
。虽然效率差了些,但还算直观,代码实现如下:
class Solution(object):
def maxSlidingWindow(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: List[int]
"""
import heapq