剑指 Offer 59 - I. 滑动窗口的最大值

题目链接:

https://leetcode-cn.com/problems/hua-dong-chuang-kou-de-zui-da-zhi-lcof/

题意:

给定一个数组 nums 和滑动窗口的大小 k,请找出所有滑动窗口里的最大值

题解:

求最大值,原本想用大顶堆去做的,但是python里好像自己没实现大顶堆只有小顶堆。所以将所有的值取负,转换为求最小值问题。
利用两个小顶堆实现:

  • 一个小顶堆用于存储区间内的最小值,一个小顶堆用于存储已不在区间内但还未从前一个小顶堆中删除的值
  • 在每次移动窗口时,向堆1中加入新加入的值,向堆2中加入需删除的值
  • 在获得当前窗口的最小值的时候,判断两个小顶堆的堆顶元素是否相同,相同则同时删除,直至不同或堆2为空
代码:
class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        if not nums:return []
        heap,remov,a= [],[],[]
        for i in nums:#取负,转换成找最小值
            a.append(-i)
        for i in range(k):
            heappush(heap,a[i])
        ans = []
        for i in range(k,len(a)+1):
            while len(remov)>0 and remov[0]== heap[0]:
                heappop(remov)
                heappop(heap)
            ans.append(-heap[0])
            if i < len(a):#特殊,最后一个区间
                heappush(remov,a[i-k])
                heappush(heap,a[i])
        return ans

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值