4.8 栈和队列-滑动窗口练习题

题目：

有一个整型数组 arr 和一个大小为 w 的窗口从数组的最左边滑到最右边,窗口每次向右边滑一个位置。 返回一个长度为n-w+1的数组res，res[i]表示每一种窗口状态下的最大值。 以数组为[4,3,5,4,3,3,6,7]，w=3为例。因为第一个窗口[4,3,5]的最大值为5，第二个窗口[3,5,4]的最大值为5，第三个窗口[5,4,3]的最大值为5。第四个窗口[4,3,3]的最大值为4。第五个窗口[3,3,6]的最大值为6。第六个窗口[3,6,7]的最大值为7。所以最终返回[5,5,5,4,6,7]。

给定整形数组arr及它的大小n，同时给定w，请返回res数组。保证w小于等于n，同时保证数组大小小于等于500。

测试样例：

[4,3,5,4,3,3,6,7],8,3

返回：[5,5,5,4,6,7]

解法：

我们必须维护当前窗口内的最大值，同时，为了当这个最大值被移出窗口时能够找到下一个最大值，当前窗口内的之后的最大值也应该被保存。假设用列表win维护这些值，应该做到：窗口始终保持右移，当一个元素arr[i]加入win时，win必须是非升序的，如果i前面有其它小于arr[i]的元素，输出窗口最大值时也不会输出它们，且它们会比第i个元素先移出窗口；当一个元素因为窗口右移退出时，如果它就是当前窗口的最大值，那么win更新为win[1:]。所以，我们可以用一个双端队列作为win的数据结构。

• 假设窗口一直右移，用l和r记录当前窗口左边界和右边界，当有arr[r]加入win时：

        win内元素从队列右端出队，直到arr[r] <= win[-1]，然后r从右端入队（注意这里入队的是元素序号）；

• 当r - l >=  w时（窗口值满）：

        比较l和win[0]，如果相等，说明上一个窗口的最大值正好该被移出窗口了，于是从左侧出队。

代码：

class SlideWindow:
    def slide(self, arr, n, w):
        # write code here
        l, r = 0, 0
        window = []
        res = []
        while r < n:
            if r-l >= w:
                if window[0] == l:
                    window = window[1:]
                l += 1
            while len(window) > 0 and arr[r] > arr[window[-1]]:
                window = window[:-1]
            window.append(r)
            if r-l == w-1:
                res.append(arr[window[0]])
            r += 1
        return res

NOTE：

这题其实我的困惑在于，什么时候应该记录窗口内的最大值，什么时候应该将l右移，什么时候可以加入新的值。看上去3者的顺序似乎不会影响结果。但是我发现，读取窗口内的最大值事实上已经隐含了“win内一定有值”这一条件，所以，不能这样写：

class SlideWindow:
    def slide(self, arr, n, w):
        # write code here
        l, r = 0, 0
        window = []
        res = []
        while r < n:
            if r-l == w-1: #如果此时window内没有值，就会报错。比如w=1的情况下
                res.append(arr[window[0]])
            r += 1
            if r-l >= w:
                if arr[window[0]] == arr[l]:
                    window = window[1:]
                l += 1
            while len(window) > 0 and r < len(arr) and arr[r] > arr[window[-1]]:
                window = window[:-1]
            window.append(r)
        return 

可以这样调整：

class SlideWindow:
    def slide(self, arr, n, w):
        # write code here
        l, r = 0, 0
        window = []
        res = []
        while r < n:
            while len(window) > 0 and arr[r] > arr[window[-1]]:
                window = window[:-1]
            window.append(r)
            if r-l == w-1:
                res.append(arr[window[0]])
            r += 1
            if r-l >= w:
                if arr[window[0]] == arr[l]:
                    window = window[1:]
                l += 1
        return res