Golangで標準ライブラリを利用した最小・最大ヒープの実装

package heap

// 便利上すべて展開して書いていますが、実際のインターフェースはsort.Interfaceを埋め込んでいます。
type Interface interface {
    Len() int           // ヒープの長さを返す
    Less(i, j int) bool // i番目の要素がj番目の要素より小さいとtrueを返す
    Swap(i, j int)      // i番目の要素とj番目の要素を交換
    Push(x any)         // ヒープに新しい要素を追加
    Pop() any           // ヒープから最小の要素を削除して返す
}

package main

import (
	"container/heap"
	"fmt"
)

// IntHeap - int型のスライスをヒープとして実装した構造体。
type IntHeap []int

// Len - ヒープの長さを返します。
func (h *IntHeap) Len() int {
	return len(*h)
}

// Less - i番目の要素がj番目の要素より小さい場合はtrueを返します。
// NOTE: ソートの基準を変更したい場合、こちらのメソッドを使用できます。
// 特定の構造体ヒープであれば、構造体の特定のフィールド：ex) `(*h)[i].Field < (*h)[j].Field`のように記述します。
func (h *IntHeap) Less(i, j int) bool {
	return (*h)[i] < (*h)[j]
}

// Swap - i番目の要素とj番目の要素を交換します。実際のヒープではUpheapとDownheapを実装するために使用されます。
func (h *IntHeap) Swap(i, j int) {
	(*h)[i], (*h)[j] = (*h)[j], (*h)[i]
}

// Push - ヒープに新しい要素を追加します。
func (h *IntHeap) Push(x any) {
	*h = append(*h, x.(int))
}

// Pop - ヒープから最小の要素を削除して返します。
func (h *IntHeap) Pop() any {
	old := *h
	n := len(old)
	x := old[n-1]
	*h = old[0 : n-1]
	return x
}

func main() {
	// IntHeapを初期化します。
	h := &IntHeap{2, 1, 5, 3, 4}
	heap.Init(h) // ヒープを初期化します。

	// ヒープに新しい要素を追加します。
	head.Push(h, 0)

	for h.Len() > 0 {
		fmt.Println(heap.Pop(h)) // ヒープから最小の要素を削除して出力します。
	}
}