如何在 Python 中实现堆？
在 Python 中实现堆，可以使用内置的 heapq 模块，它提供了许多有用的函数来实现堆。如下面的代码演示：
import heapq # 将列表转换为堆 heap = [3, 1, 4, 1, 5, 9, 2, 6, 5, 3] heapq.heapify(heap) # 向堆中添加元素 heapq.heappush(heap, 0) # 从堆中弹出最小值 min_val = heapq.heappop(heap) # 获取堆中最小的元素，但不删除它 min_val = heap[0] # 获取 n 个最小的元素 n_smallest = heapq.nsmallest(n, heap) # 获取 n 个最大的元素 n_largest = heapq.nlargest(n, heap)
上面的代码演示了如何使用 heapq 来实现堆，其中 heapq.heapify 函数将普通列表转换为堆，heapq.heappush 函数向堆中添加元素，heapq.heappop 函数从堆中弹出最小值，heap[0] 可以获取堆中最小的元素但不删除它。heapq.nsmallest 和 heapq.nlargest 函数可以获取 n 个最小的或最大的元素。
如果要将字符串作为堆中元素的范例，可以稍微修改上面的代码，如下所示：
import heapq # 将字符串列表转换为堆 heap = ["pidancode.com", "Python", "heapq", "heap", "sort", "algorithm"] heapq.heapify(heap) # 向堆中添加字符串元素 heapq.heappush(heap, "皮蛋编程") # 从堆中弹出最小值 min_val = heapq.heappop(heap) # 获取堆中最小的元素，但不删除它 min_val = heap[0] # 获取 n 个最小的元素 n_smallest = heapq.nsmallest(n, heap) # 获取 n 个最大的元素 n_largest = heapq.nlargest(n, heap)
上面的代码演示了如何使用 heapq 模块来实现字符串堆。
相关文章