0

0

Python 优先队列的实现思路

冰川箭仙

冰川箭仙

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

|

797人浏览过

|

来源于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免费学习笔记(深入)”;

hstshop鸿思特商城系统
hstshop鸿思特商城系统

鸿思特商城系统HstShop是一款B2C独立网店系统,由拥有十年互联网开发经验的牛头带队开发完成,完全免费开源,适合大中型网站平台快速构建立强大的网上商城平台网店系统。HstShop悉心听取每一位商家的需求与建议,根据中国人的购物习惯改进了购物流程,实现更好的用户购物体验。HstShop网店系统无论在产品功能、稳定性、执行效率、负载能力、安全性和搜索引擎优化等方面都居国内同类产品领先地位,成为国内

下载
  • 数值型:存 -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工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

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

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

211

2023.10.12

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

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

531

2023.08.03

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

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

214

2023.09.04

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

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

1552

2023.10.24

字符串介绍
字符串介绍

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

640

2023.11.24

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

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

885

2024.03.22

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

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

836

2024.04.29

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

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

185

2025.07.29

pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法
pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法

本专题系统整理pixiv网页版官网入口及登录访问方式,涵盖官网登录页面直达路径、在线阅读入口及快速进入方法说明,帮助用户高效找到pixiv官方网站,实现便捷、安全的网页端浏览与账号登录体验。

76

2026.02.13

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 4.3万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.5万人学习

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号