在Golang的标准库中，container/heap包提供了堆结构的通用实现方案，开发者不需要手动实现堆的底层调整逻辑，只需要自定义一个类型并实现heap.Interface接口中定义的方法，就能快速构建符合需求的堆结构，其中最小堆和最大堆是两种最常用的堆类型。
container/heap核心接口说明
要使用container/heap实现堆，首先需要了解heap.Interface的定义，它继承自sort.Interface，包含以下必须实现的方法：
- Len() int：返回集合中元素的数量
- Less(i, j int) bool：判断索引i的元素是否小于索引j的元素，这个方法决定了堆的类型
- Swap(i, j int)：交换索引i和j位置的元素
- Push(x interface{})：向集合末尾添加一个元素
- Pop() interface{}：移除并返回集合末尾的元素
实现上述所有方法后，就可以调用heap.Init初始化堆，调用heap.Push和heap.Pop进行元素的插入和删除操作。
实现最小堆示例
最小堆的特点是父节点的值总是小于等于子节点的值，根节点是集合中的最小值。我们可以通过Less方法的实现来定义最小堆的比较逻辑。
最小堆类型定义与方法实现
package main
import (
"container/heap"
"fmt"
)
// 定义最小堆类型，底层使用int切片存储元素
type MinHeap []int
// 实现Len方法，返回堆中元素数量
func (h MinHeap) Len() int {
return len(h)
}
// 实现Less方法，i位置元素小于j位置元素时返回true，这是最小堆的核心逻辑
func (h MinHeap) Less(i, j int) bool {
return h[i] < h[j]
}
// 实现Swap方法，交换两个位置的元素
func (h MinHeap) Swap(i, j int) {
h[i], h[j] = h[j], h[i]
}
// 实现Push方法，向堆末尾添加元素
func (h *MinHeap) Push(x interface{}) {
*h = append(*h, x.(int))
}
// 实现Pop方法，移除并返回堆末尾的元素
func (h *MinHeap) Pop() interface{} {
old := *h
n := len(old)
x := old[n-1]
*h = old[0 : n-1]
return x
}
func main() {
// 初始化最小堆
h := &MinHeap{3, 1, 4, 1, 5, 9, 2, 6}
heap.Init(h)
fmt.Println("初始化后的最小堆：", *h)
// 向堆中插入元素
heap.Push(h, 0)
fmt.Println("插入0后的堆：", *h)
// 弹出堆顶元素（最小值）
min := heap.Pop(h)
fmt.Println("弹出的堆顶元素：", min)
fmt.Println("弹出后剩余的堆：", *h)
}
最小堆代码说明
上述代码中，MinHeap的Less方法判断h[i]是否小于h[j]，符合最小堆的比较规则。main函数中首先初始化堆，然后演示了插入元素和弹出堆顶元素的操作，每次操作后heap包会自动调整堆结构，保证堆的性质。
实现最大堆示例
最大堆的特点是父节点的值总是大于等于子节点的值，根节点是集合中的最大值，只需要修改Less方法的实现逻辑即可。
最大堆类型定义与方法实现
package main
import (
"container/heap"
"fmt"
)
// 定义最大堆类型，底层使用int切片存储元素
type MaxHeap []int
// 实现Len方法
func (h MaxHeap) Len() int {
return len(h)
}
// 实现Less方法，i位置元素大于j位置元素时返回true，这是最大堆的核心逻辑
func (h MaxHeap) Less(i, j int) bool {
return h[i] > h[j]
}
// 实现Swap方法
func (h MaxHeap) Swap(i, j int) {
h[i], h[j] = h[j], h[i]
}
// 实现Push方法
func (h *MaxHeap) Push(x interface{}) {
*h = append(*h, x.(int))
}
// 实现Pop方法
func (h *MaxHeap) Pop() interface{} {
old := *h
n := len(old)
x := old[n-1]
*h = old[0 : n-1]
return x
}
func main() {
// 初始化最大堆
h := &MaxHeap{3, 1, 4, 1, 5, 9, 2, 6}
heap.Init(h)
fmt.Println("初始化后的最大堆：", *h)
// 向堆中插入元素
heap.Push(h, 10)
fmt.Println("插入10后的堆：", *h)
// 弹出堆顶元素（最大值）
max := heap.Pop(h)
fmt.Println("弹出的堆顶元素：", max)
fmt.Println("弹出后剩余的堆：", *h)
}
最大堆代码说明
最大堆的Less方法判断h[i]是否大于h[j]，这样堆调整时会把更大的元素放到父节点位置。同样的初始化、插入、弹出操作和最小堆的调用方式完全一致，只是内部比较逻辑不同，最终得到的是最大堆结构。
堆操作的注意事项
在使用container/heap时需要注意以下几点：
- 自定义堆类型的Push和Pop方法需要操作指针接收者，因为这两个方法会修改切片的内容
- 初始化堆时必须调用heap.Init，否则堆结构可能不符合预期
- 插入元素要使用heap.Push而不是直接调用自定义类型的Push方法，否则不会触发堆调整
- 弹出元素要使用heap.Pop而不是直接调用自定义类型的Pop方法，否则不会触发堆调整
通过上述示例可以看出，基于container/heap实现最小堆和最大堆的核心差异仅在于Less方法的实现逻辑，其他接口方法的实现基本一致，开发者可以根据实际需求灵活调整比较规则，实现自定义优先级的堆结构。
container_heapGolang最小堆最大堆堆结构修改时间：2026-07-09 05:24:12