【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
相关推荐
山海不说话1 小时前
视频行为标注工具BehaviLabel(源码+使用介绍+Windows.Exe版本)
人工智能·python·计算机视觉·视觉检测
门前云梦2 小时前
《C语言·源初法典》---C语言基础(上)
c语言·开发语言·学习
liuzhenghua663 小时前
Python任务调度模型
java·运维·python
小前端大牛马3 小时前
java教程笔记(十一)-泛型
java·笔记·python
sjtu_cjs3 小时前
Tensorrt python api 10.11.0笔记
开发语言·笔记·python
0x7CF3 小时前
QT3D学习笔记——圆台、圆锥
qt·学习·3d
哆啦A梦的口袋呀3 小时前
深入理解系统:UML类图
开发语言·python·uml
虎冯河4 小时前
怎么让Comfyui导出的图像不包含工作流信息,
开发语言·python
qq_386322694 小时前
华为网路设备学习-24(路由器OSPF - 特性专题)
学习·华为·智能路由器