Python - 优先队列（queue.PriorityQueue & heapq）

如果你正在学习Python，但是找不到方向的话可以试试我这一份学习方法和籽料呀！点击 领取籽料（不要米米）

Python 中实现优先队列主要有两种方式：使用标准库中的 queue.PriorityQueue 类，或者使用 heapq 模块。两者各有特点，适用于不同的场景。

1. 使用 queue.PriorityQueue

queue.PriorityQueue 是一个线程安全的优先队列实现，适用于多线程编程环境。它提供了基本的入队（put）和出队（get）操作，元素按照优先级顺序被移除（优先级最低的元素首先被移除）。

基本用法：

import queue  
  
# 创建一个优先队列  
pq = queue.PriorityQueue()  
  
# 入队，元素形式为 (priority, item)  
pq.put((2, 'b'))  
pq.put((1, 'a'))  
pq.put((3, 'c'))  
  
# 出队，返回并移除优先级最低的元素  
while not pq.empty():  
    print(pq.get())  # 输出: (1, 'a') (2, 'b') (3, 'c')

2. 使用heapq模块

heapq 模块提供了一个堆队列算法的实现，也称为优先队列算法。heapq 是一个列表，但它被当作堆来对待，提供了堆操作的方法，如堆排序、添加元素等。不过，heapq 模块本身不提供线程安全支持，适合单线程环境。

基本用法：

import heapq  
  
# 创建一个堆（优先队列）  
pq = []  
  
# heapq.heappush 用于向堆中添加元素  
heapq.heappush(pq, (2, 'b'))  
heapq.heappush(pq, (1, 'a'))  
heapq.heappush(pq, (3, 'c'))  
  
# heapq.heappop 用于从堆中移除并返回优先级最低的元素  
while pq:  
    print(heapq.heappop(pq))  # 输出: (1, 'a') (2, 'b') (3, 'c')

比较

• 线程安全：queue.PriorityQueue 是线程安全的，适合多线程环境；而 heapq 不是线程安全的，适合单线程环境。
• 功能：queue.PriorityQueue 提供了较为丰富的接口，如 put、get、qsize、empty 等；heapq 模块提供了堆操作的基本方法，如 heappush、heappop、heappushpop、heapify 等，更灵活但也需要更多的手动管理。
• 性能：在单线程环境下，heapq 由于其底层实现（基于列表）可能在一些操作上（如插入和删除）比 queue.PriorityQueue（基于堆队列）更高效。但在多线程环境下，queue.PriorityQueue 的线程安全特性可能会引入一些性能开销。

选择哪种方式取决于你的具体需求，如是否需要线程安全、是否需要更多的控制以及性能考虑等。