LeetCode 堆的插入与删除题解

LeetCode 堆的插入与删除题解

题目描述

实现堆的插入和删除操作。

堆的插入与删除

插入操作

1. 将新元素添加到数组末尾
2. 向上调整（sift up）

删除操作

1. 删除堆顶元素
2. 将数组末尾元素移到堆顶
3. 向下调整（sift down）

代码实现

import heapq class MinHeap: def __init__(self): self.heap = [] def push(self, val): heapq.heappush(self.heap, val) def pop(self): return heapq.heappop(self.heap) def top(self): return self.heap[0] if self.heap else None # 测试 def test_min_heap(): heap = MinHeap() heap.push(3) heap.push(1) heap.push(2) print(heap.pop()) # 输出：1 print(heap.top()) # 输出：2 if __name__ == "__main__": test_min_heap()

总结

堆的插入和删除操作是堆的基本操作，通过向上和向下调整来维护堆的性质。