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)
	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])

	// 移除并返回堆顶元素。
	pop := heap.Pop(h).(int)
	fmt.Println("弹出的元素:", pop)
	fmt.Println("最小元素，在移除元素之后:", (*h)[0])
}

最小元素: 1
最小元素，在添加 3 之后: 1
弹出的元素: 1
最小元素，在移除元素之后: 2