双端队列解决滑动窗口问题Python

2018-03-08  本文已影响159人  Crystalajj
class SlideWindow:
    def slide(self, arr, n, w):
        qmax = []  #双端队列,队头记录最大值的下标
        res = []
        for i in range(n):
            while qmax != []:
                j = qmax[-1]
                if arr[j] > arr[i]:
                    break
                else:
                    qmax.pop()
            qmax.append(i)
            if i >= w-1:
                if i-qmax[0]>=w:
                    qmax.pop(0)
                res.append(arr[qmax[0]])
        return res

上一篇 下一篇

猜你喜欢

热点阅读