LeetCode 239. 滑动窗口最大值

原题地址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 ≤ 输入数组的大小，且输入数组不为空。

进阶：

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

---

算法

队列 deque 记录极大值的下标。逐个遍历，对当前点 nums[i] ，若不比队尾大则进入 deque ；否则，队尾出队直到队尾不小于  nums[i] ， nums[i] 入队。

若队头的下标和 i 的距离超过 k ，即位于窗口之外，出队。

当 i 不小于 k-1 的时候，就需要输出队头元素。

---

代码

def fun(nums, k):
    """
    :type nums: List[int]
    :type k: int
    :rtype: List[int]
    """
    ans = []
    deque = []
    for i in range(len(nums)):
        while len(deque) > 0 and nums[deque[-1]] < nums[i]:
            deque.pop()
        deque.append(i)
        if i >= deque[0] + k:
            deque.pop(0)
        if i >= k - 1:
            ans.append(nums[deque[0]])
    return ans