Go最小堆

package main

import (
	"container/heap"
	"fmt"
)

// IntHeap 是一个包含整数的切片，它实现了 heap.Interface 接口。
type IntHeap []int

// Len 方法返回堆中的元素数量。
func (h IntHeap) Len() int { return len(h) }

// Less 方法报告索引 i 的元素是否比索引 j 的元素小。
func (h IntHeap) Less(i, j int) bool { return h[i] < h[j] }

// Swap 方法交换索引 i 和 j 处的元素。
func (h IntHeap) Swap(i, j int) { h[i], h[j] = h[j], h[i] }

// Push 方法将元素 x 添加到堆的最后。
func (h *IntHeap) Push(x interface{}) {
	*h = append(*h, x.(int))
}

// Pop 方法从堆中取出最后一个元素，并返回它。
func (h *IntHeap) Pop() interface{} {
	old := *h
	n := len(old)
	if n == 0 {
		// 或者可以返回一个错误值，这里选择返回nil
		return nil
	}
	x := old[n-1]
	*h = old[0 : n-1]
	return x
}

func main() {
	// 创建一个最小堆实例。
	h := &IntHeap{2, 1, 5}
	heap.Init(h)
	fmt.Println("最小元素:", (*h)[0]) // h[0] 是最小元素

	// 添加一个新元素到堆中。
	heap.Push(h, 3)
	fmt.Println("最小元素，在添加 3 之后:", (*h)[0])

	// 弹出堆顶元素直到堆为空。
	for h.Len() > 0 {
		pop := heap.Pop(h).(int)
		fmt.Println("弹出的元素:", pop)
	}

	// 检查堆是否为空。
	if h.Len() == 0 {
		fmt.Println("堆已为空，没有更多元素可以弹出。")
	} else {
		// 堆不为空，尝试弹出元素并处理异常。
		pop := heap.Pop(h).(int)
		fmt.Println("弹出的元素:", pop)
	}
}

最小元素: 1
最小元素，在添加 3 之后: 1
弹出的元素: 1
弹出的元素: 2
弹出的元素: 3
弹出的元素: 5
堆已为空，没有更多元素可以弹出。