Leetcode-滑动窗口的最大值_python

239.给定一个数组 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 ≤ 输入数组的大小,且输入数组不为空。

进阶:

你能在线性时间复杂度内解决此题吗?

思路:

1.根据优先队列的概念,我们假设一个大顶堆,那么一开始的[1,3,-1],这样一排列成堆的样子就是3在最上面,-1在左下角,1在右下角。。下一步就是[3,-1,-3]了,1就要被挤开了,挤开了也不影响什么,-3再加进来就好了。总之我们需要做的是:(1)维护我们的Heap,也就是删除离开窗口的元素,加入新的元素。这里时间复杂度是logK (2)Max->Top,就是让结果是堆顶的元素。复杂度是O(1),最后整体的复杂度是NLogK。有没有更好的解法?

2.直接用队列,而且是双端队列,也就是两边都能进能出的队列。首先就是入队列,每次滑动窗口都把最大值左边小的数给杀死,也就是出队,后面再滑动窗口进行维护,这样相当于就是每一个数走过场,时间复杂度就是O(N*1),比思路1要小。

class Solution:
    def maxSlidingWindow(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[int]
        """
        #严谨判断输入的数字是否合法
        if not nums:return []
        window, res = [], []
        for i, x in enumerate(nums):
            if i>=k and window[0] <= i-k: #窗口滑动时的规律
                window.pop(0)
            while window and nums[window[-1]] <= x: #把最大值左边的数小的就清除。
                window.pop()
            window.append(i)
            if i >= k-1:
                res.append(nums[window[0]])
        return res

你可能感兴趣的:(LeetCode,算法与数据结构)