0

0

Python中heapq模块 堆队列算法heapq的优先级队列实现

冰火之心

冰火之心

发布时间:2025-06-10 16:57:03

|

827人浏览过

|

来源于php中文网

原创

python中,使用heapq模块可以很方便地实现优先级队列。堆(heap)是一种特殊的树结构,常用于快速获取最小值或最大值的场景。heapq默认实现的是最小堆,也就是说堆顶元素始终是整个堆中最小的那个。

什么是优先级队列?

优先级队列是一种数据结构,每个元素都带有一个“优先级”。当访问或删除元素时,优先级更高的元素会优先被处理。比如任务调度、医院挂号系统等场景都会用到这种结构。

常见的做法是将每个元素和它的优先级打包成一个元组,然后插入堆中。Python的heapq会根据元组的第一个元素进行排序,从而实现优先级控制。


如何用heapq创建优先级队列

使用heapq实现优先级队列非常简单,只需要导入模块并维护一个堆列表即可:

  • 使用heapq.heappush()添加元素
  • 使用heapq.heappop()弹出当前优先级最高的元素(即最小的)

例如:

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

import heapq

queue = []
heapq.heappush(queue, (3, '写报告'))
heapq.heappush(queue, (1, '回邮件'))
heapq.heappush(queue, (2, '开会'))

print(heapq.heappop(queue))  # 输出:(1, '回邮件')

每次调用heappop()都会取出当前优先级最高的任务。

注意:如果两个元素的优先级相同(比如都是2),那么它们之间的顺序取决于后面的元素内容,在heapq中不保证稳定排序。

处理优先级相同的情况

当多个任务拥有相同的优先级时,你可能还希望这些任务能按照它们插入的顺序来处理。这时候可以在元组中加入第三个字段,比如时间戳:

魔术橡皮擦
魔术橡皮擦

智能擦除、填补背景内容

下载
import heapq
import time

queue = []
timestamp = 0

def add_task(priority, task):
    global timestamp
    heapq.heappush(queue, (priority, timestamp, task))
    timestamp += time.time()

add_task(2, '任务A')
add_task(2, '任务B')
add_task(1, '紧急任务')

print(heapq.heappop(queue))  # 紧急任务先出
print(heapq.heappop(queue))  # A比B早添加,所以先出
print(heapq.heappop(queue))  # 最后才是B

这样即使优先级相同,也能通过时间戳保持插入顺序。


常见问题与注意事项

  • heapq默认是最小堆,如果你需要最大堆行为,可以把优先级取负数再插入。

    heapq.heappush(queue, (-priority, task))
  • 不要直接操作堆列表,比如用append()而不是heappush(),这会导致堆结构损坏。

  • heapq适用于中小型数据量。如果数据特别大或者频繁操作,要考虑其他结构如线段树、平衡树等。


基本上就这些。heapq虽然简单,但配合元组使用就能轻松实现优先级队列功能,适合很多实际场景。只要注意好插入和弹出的方式,以及优先级冲突的处理方式,基本不会出错。

相关专题

更多
python开发工具
python开发工具

php中文网为大家提供各种python开发工具,好的开发工具,可帮助开发者攻克编程学习中的基础障碍,理解每一行源代码在程序执行时在计算机中的过程。php中文网还为大家带来python相关课程以及相关文章等内容,供大家免费下载使用。

720

2023.06.15

python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

627

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

744

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

617

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

1236

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

547

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

575

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

700

2023.08.11

php源码安装教程大全
php源码安装教程大全

本专题整合了php源码安装教程,阅读专题下面的文章了解更多详细内容。

74

2025.12.31

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
PHP自制框架
PHP自制框架

共8课时 | 0.6万人学习

简单聊聊mysql8与网络通信
简单聊聊mysql8与网络通信

共1课时 | 779人学习

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

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