美文网首页
双端队列解决滑动窗口问题Python

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

作者: Crystalajj | 来源:发表于2018-03-08 20:45 被阅读159次
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

相关文章

网友评论

      本文标题:双端队列解决滑动窗口问题Python

      本文链接:https://www.haomeiwen.com/subject/cmtufftx.html