Python堆

堆是一种特殊的树结构，其中每个父节点小于或等于其子节点。 然后它被称为最小堆(Min Heap)。 如果每个父节点大于或等于其子节点，则称它为最大堆(Max Heap)。 实施优先级队列是非常有用的，在该队列中，具有较高权重的队列项目在处理中具有更高的优先级。在本章中，我们将学习使用python实现堆数据结构。

创建一个堆

堆是通过使用python内建的名称为heapq的库创建的。 该库具有对堆数据结构进行各种操作的相关功能。 以下是这些函数的列表 -

• heapify - 此函数将常规列表转换为堆。 在结果堆中，最小的元素被推到索引位置0。但是其余的数据元素不一定被排序。
• heappush - 这个函数在堆中添加一个元素而不改变当前堆。
• heappop - 该函数返回堆中最小的数据元素。
• heapreplace - 该函数用函数中提供的新值替换最小的数据元素。

通过简单地使用具有heapify函数的元素列表来创建堆。 在下面的例子中，提供了一个元素列表，heapify函数重新排列了元素到最初位置的元素。

import heapq

H = [21,1,45,78,3,5]
# Use heapify to rearrange the elements
heapq.heapify(H)
print(H)

执行上面示例代码，得到以下结果 -

[1, 3, 5, 78, 21, 45]

插入堆

将数据元素插入堆总是在最后一个索引处添加元素。 但是，只有在值最小的情况下，才可以再次应用heapify函数将新添加的元素添加到第一个索引。 在下面的例子中，插入数字 - 8 。

import heapq
H = [21,1,45,78,3,5]
# Covert to a heap
heapq.heapify(H)
print(H)
# Add element
heapq.heappush(H,8)
print(H)

执行上面示例代码，得到以下结果 -

[1, 3, 5, 78, 21, 45]
[1, 3, 5, 78, 21, 45, 8]

从堆中移除

可以使用此功能在第一个索引处移除元素。 在下面的例子中，函数将始终删除索引位置1处的元素。

import heapq

H = [21,1,45,78,3,5]
# Create the heap

heapq.heapify(H)
print(H)

# Remove element from the heap
heapq.heappop(H)

print(H)

执行上面示例代码，得到以下结果 -

[1, 3, 5, 78, 21, 45]
[3, 21, 5, 78, 45]

替换堆

heapreplace函数总是删除堆中最小的元素，并在未被任何顺序修复的地方插入新的传入元素。参考以下示例 -

import heapq

H = [21,1,45,78,3,5]
# Create the heap

heapq.heapify(H)
print(H)

# Replace an element
heapq.heapreplace(H,6)
print(H)

执行上面示例代码，得到以下结果 -

[1, 3, 5, 78, 21, 45]
[3, 6, 5, 78, 21, 45]