数据结构-堆 - 技术栈

数据结构-堆

How_doyou_do2025-05-15 18:42

堆是一颗树，其每个节点都有一个值，且（小根堆：每个父节点都小于等于其子节点

（在进树时不断进行比较

STL中的 priority_queue 其实就是一个大根堆

python中可以用 heapq（小根堆）和 PriorityQueue （大根堆）

heapq

heapq默认是小根堆，如果想实现大根堆的话得负负得正

python 复制代码

import heapq

python 复制代码

heap=[]

或者从列表转化而来

python 复制代码

heapq.heapify(列表名)

那么后面堆名就是列表名

python 复制代码

heapq.heappush(堆名，插入元素值)

python 复制代码

heapq.heappop(堆名)

在这PQ里我们需要自己设置优先级，一般情况下就是元素值等于优先级值

注意⚠️：PriorityQueue得从不是标准库的deque导入

python 复制代码

from deque import PriorityQueue

python 复制代码

pq=PriorityQueue()

python 复制代码

pq.put([优先级，元素值])

取出优先级最高的（小根堆的话也是负负得正

python 复制代码

pq.get()