【python】优先队列-堆 学习 Leetcode 239. 滑动窗口最大值

给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。

返回 滑动窗口中的最大值 。

示例 1:

输入:nums = [1,3,-1,-3,5,3,6,7], k = 3

输出:[3,3,5,5,6,7]

示例 2:

输入:nums = [1], k = 1

输出:[1]

提示:

1 <= nums.length <= 105

-104 <= nums[i] <= 104

1 <= k <= nums.length

python使用heapq。注意python的heapq在技术上只实现了小顶堆,因此传入的参考值应取负以实现大顶堆。

代码:

python 复制代码
class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        # 小顶堆法
        ret_arr = []
        heap = []
        for i in range(k-1):
            heapq.heappush(heap, (-nums[i],i))
        
        for i in range(len(nums)-k+1):
            heapq.heappush(heap, (-nums[i+k-1],i+k-1))
            while(heap[0][1] < i): heapq.heappop(heap)
            ret_arr.append(-heap[0][0])
                
        return ret_arr
相关推荐
Dxy12393102161 天前
python把文件从一个文件复制到另一个文件夹
开发语言·python
sonrisa_1 天前
collections模块
python
折翼的恶魔1 天前
数据分析:排序
python·数据分析·pandas
兔子不吃草~1 天前
Transformer学习记录与CNN思考
学习·cnn·transformer
天雪浪子1 天前
Python入门教程之赋值运算符
开发语言·python
站大爷IP1 天前
5个技巧写出专业Python代码:从新手到进阶的实用指南
python
_不会dp不改名_1 天前
leetcode_21 合并两个有序链表
算法·leetcode·链表
hrrrrb1 天前
【Python】字符串
java·前端·python
与己斗其乐无穷1 天前
C++学习记录(8)list
学习
有谁看见我的剑了?1 天前
k8s-容器探针和生命周期回调学习
学习·容器·kubernetes