Python - 堆

堆是一种特殊的树结构，其中每个父节点小于或等于其子节点。然后它被称为Min 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]