讲师中心 微信公众号
AI工具推荐 视频效率加速

Python 优先队列的实现思路

风强小哥_1857

风强小哥_1857

发布时间:2026-02-14 22:34:03

|

837人浏览过

|

来源于php中文网

原创

Python中heapq是优先队列的底层实现,通过heappush/heappop维护最小堆结构;需用取反或元组实现最大堆;单线程推荐直接用heapq,多线程才用queue.PriorityQueue。

python 优先队列的实现思路

Python 里 heapq 就是优先队列的底层实现

Python 标准库没有叫 PriorityQueue 的类来直接当“队列”用,但 heapq 模块提供了一组原地操作最小堆的函数,这才是实际干活的核心。你不是在“创建一个队列对象”,而是在维护一个列表,靠 heapq.heappush() 和 heapq.heappop() 保证它始终满足堆序。

常见错误是以为 heapq 自动排序整个列表——其实它只维护堆结构,不保证列表其余部分有序;还有人误用 sorted() 替代 heappop(),结果每次取最小值都 O(n log n),而不是 O(log n)。

  • heapq.heappush(heap, item):把 item 插入 heap 并调整结构,时间复杂度 O(log n)
  • heapq.heappop(heap):弹出并返回最小元素,同时修复堆,O(log n)
  • 初始化堆用 heapq.heapify(lst),比逐个 heappush 快,O(n)
  • 不要手动修改 heap 列表(比如 heap[0] = x),会破坏堆性质

需要最大堆?别改源码,改数据就行

Python 的 heapq 只支持最小堆,但多数场景要的是“最大优先级先出”。最稳妥的做法不是重写堆逻辑,而是对数值取反或包装成元组,让比较逻辑倒过来。

比如你要按分数降序取学生:heapq 默认按第一个元素升序,所以把 (-score, student_id) 入堆,heappop 后再取负号还原即可。字符串或自定义对象也类似,关键在于让 __lt__ 行为符合你的“优先级高低”定义。

立即学习“Python免费学习笔记(深入)”;

提示词大师-python版
提示词大师-python版

图片提示词生成器?不止如此。 马甲系统 —— 把脑海中的画面,翻译成AI能理解的专业表达。 用得越多,它越懂你:首次需要多问几句确认方向,用久了几乎一说就懂。 用得越多,它越快:缓存机制让后续对话越来越省。 RAG进化:成功案例持续入库,越跑越聪明。 输入「新手指南」查看完整功能介绍

下载
  • 数值型:存 -x,取出来再 -item
  • 多字段:用元组 (-priority, timestamp, item),确保可比且稳定
  • 避免用 max heap = heapq._heapify_max:这是私有 API,不保证兼容,Python 3.12 已明确标记为内部使用
  • 如果频繁需要最大堆,封装一层类比裸用 heapq 更安全,但别过度设计

queue.PriorityQueue 是线程安全的包装,别在单线程里滥用

标准库里的 queue.PriorityQueue 确实名字像“正统优先队列”,但它本质是用 heapq 加了锁的封装,目标是多线程生产者-消费者场景。如果你在单线程脚本、算法题或数据处理中用它,反而引入不必要的开销和 API 隔离(比如不能直接访问底层列表)。

典型误用:用 PriorityQueue.put(x) 和 PriorityQueue.get() 写 Dijkstra,结果发现没法做“减小键值”(decrease-key)操作,也没法检查某个节点是否已在队列中——因为它的接口不暴露内部状态。

  • 单线程、需灵活控制(如更新权重、查存在性)→ 直接用 heapq + 搭配字典或集合
  • 多线程、只需“塞进去、取出来”且不关心中间状态 → queue.PriorityQueue 合理
  • queue.PriorityQueue 不支持 len() 以外的长度外查,也不支持索引访问
  • 它的 put() 和 get() 是阻塞式,默认会等锁,哪怕队列非空

自定义比较逻辑时,__lt__ 是唯一靠谱入口

想让类实例能进 heapq,必须定义 __lt__(less-than)。Python 堆操作只调这个方法做比较,其他如 __eq__ 或 __gt__ 完全无关。很多人试了 __cmp__(Python 2)或试图重载 >,结果报 TypeError: '。

注意:如果两个对象 __lt__ 返回 False,不代表它们相等,可能只是“不小于”,堆操作仍可能出错。稳妥做法是让 __lt__ 总返回布尔值,并确保逻辑可传递、无歧义。

  • 只实现 __lt__,别碰 __le__ 或 __gt__,heapq 不用它们
  • 元组天然可比,所以 (priority, insertion_count, obj) 是避免自定义类的好技巧
  • 如果 priority 相同,不加 insertion_count 可能导致不可预测的顺序(取决于对象 id 比较,而它不保证稳定)
  • 别在 __lt__ 里做耗时计算或 I/O,堆操作本身高频,慢逻辑会拖垮性能
事情说清了就结束。真正卡住人的,往往不是“怎么建堆”,而是“怎么让不同优先级的数据在堆里不打架”,以及“什么时候该放弃封装去碰底层列表”。

热门AI工具

更多
火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

相关专题

更多
Sass和less的区别
Sass和less的区别

Sass和less的区别有语法差异、变量和混合器的定义方式、导入方式、运算符的支持、扩展性等。本专题为大家提供Sass和less相关的文章、下载、课程内容,供大家免费下载体验。

2247

2023.10.12

js 字符串转数组
js 字符串转数组

js字符串转数组的方法:1、使用“split()”方法;2、使用“Array.from()”方法;3、使用for循环遍历;4、使用“Array.split()”方法。本专题为大家提供js字符串转数组的相关的文章、下载、课程内容,供大家免费下载体验。

1578

2023.08.03

js截取字符串的方法
js截取字符串的方法

js截取字符串的方法有substring()方法、substr()方法、slice()方法、split()方法和slice()方法。本专题为大家提供字符串相关的文章、下载、课程内容,供大家免费下载体验。

2284

2023.09.04

java基础知识汇总
java基础知识汇总

java基础知识有Java的历史和特点、Java的开发环境、Java的基本数据类型、变量和常量、运算符和表达式、控制语句、数组和字符串等等知识点。想要知道更多关于java基础知识的朋友,请阅读本专题下面的的有关文章,欢迎大家来php中文网学习。

5804

2023.10.24

字符串介绍
字符串介绍

字符串是一种数据类型,它可以是任何文本,包括字母、数字、符号等。字符串可以由不同的字符组成,例如空格、标点符号、数字等。在编程中,字符串通常用引号括起来,如单引号、双引号或反引号。想了解更多字符串的相关内容,可以阅读本专题下面的文章。

4929

2023.11.24

java读取文件转成字符串的方法
java读取文件转成字符串的方法

Java8引入了新的文件I/O API,使用java.nio.file.Files类读取文件内容更加方便。对于较旧版本的Java,可以使用java.io.FileReader和java.io.BufferedReader来读取文件。在这些方法中,你需要将文件路径替换为你的实际文件路径,并且可能需要处理可能的IOException异常。想了解更多java的相关内容,可以阅读本专题下面的文章。

6634

2024.03.22

php中定义字符串的方式
php中定义字符串的方式

php中定义字符串的方式:单引号;双引号;heredoc语法等等。想了解更多字符串的相关内容,可以阅读本专题下面的文章。

8994

2024.04.29

go语言字符串相关教程
go语言字符串相关教程

本专题整合了go语言字符串相关教程,阅读专题下面的文章了解更多详细内容。

4559

2025.07.29

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

160

2026.09.23

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn