给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回滑动窗口中的最大值。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sliding-window-maximum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution: def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]: n=len(nums) q=[(-nums[i],i)for i in range(k)]#找到前k个并且取负数,记录下标值 heapq.heapify(q)#将列表建成小顶堆 ans=[-q[0][0]]#记录顶值,取负回来 for i in range(k,n): heapq.heappush(q,(-nums[i],i))#将下一个拿入 while q[0][1]<=i-k: heapq.heappop(q)#剔除顶 ans.append(-q[0][0])#否则就把它给加入 return ans
原文:https://www.cnblogs.com/liuxiangyan/p/14589308.html