Python数据结构之优先级队列queue用法详解

什么是优先级队列？

优先级队列是一种特殊的队列，它的每个元素都有一个与之关联的优先级。当元素加入队列时，会根据优先级进行排序，优先级最高的元素会排在队列的前面。当需要取出元素时，会先取出优先级最高的元素。

Python中的优先级队列模块queue

Python的标准库中提供了一个优先级队列模块queue，该模块提供了PriorityQueue类，用于实现优先级队列。

PriorityQueue类的常用方法

1. put 方法

put(item, priority)方法用于将元素加入队列。item参数指定要加入队列的元素，priority参数指定元素的优先级。优先级可以是任何可比较的对象，比如数字或字符串。

示例：

from queue import PriorityQueue

q = PriorityQueue()
q.put('C', 3)
q.put('A', 1)
q.put('B', 2)

while not q.empty():
    item = q.get()
    print(item)

输出：

A
B
C

2. get 方法

get()方法用于从队列中取出优先级最高的元素并返回。如果队列为空，则会一直等待直到队列中有新的元素为止。

示例：

from queue import PriorityQueue

q = PriorityQueue()
q.put('C', 3)
q.put('A', 1)
q.put('B', 2)

print(q.get())  # 输出：A
print(q.get())  # 输出：B
print(q.get())  # 输出：C

3. empty 方法

empty()方法用于判断队列是否为空。如果队列为空，则返回True；否则返回False。

示例：

from queue import PriorityQueue

q = PriorityQueue()

print(q.empty())  # 输出：True

q.put('A', 1)

print(q.empty())  # 输出：False

4. qsize 方法

qsize()方法用于获取队列的大小，即队列中当前元素的个数。

示例：

from queue import PriorityQueue

q = PriorityQueue()
q.put('A', 1)
q.put('B', 2)

print(q.qsize())  # 输出：2

其他扩展功能

除了上述基本方法外，PriorityQueue类还提供了一些其他有用的方法，例如task_done()和join()，用于线程间的协同工作。详情可参考Python官方文档。

以上就是Python数据结构之优先级队列queue的用法详解。希望能对你有所帮助！