Golang 中的大顶堆或小顶堆

Golang 中没有提供可直接使用的大顶堆或小顶堆，需要自己去实现 container/heap 包中的 heap.Interface 接口才能实现，具体如下。

package main

import (
    "container/heap"
    "fmt"
)

type IntHeap []int

func (h IntHeap) Len() int           { return len(h) }
func (h IntHeap) Swap(i, j int)      { h[i], h[j] = h[j], h[i] }
func (h IntHeap) Less(i, j int) bool { return h[i] > h[j] }   // 大顶堆
//func (h IntHeap) Less(i, j int) bool { return h[i] < h[j] } // 小顶堆

func (h *IntHeap) Push(x interface{}) {
    *h = append(*h, x.(int))
}

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{3, 1, 2, 5}
    heap.Init(h)
    heap.Push(h, 4)
    fmt.Println(heap.Pop(h))
    for h.Len() > 0 {
        fmt.Printf("%d ", heap.Pop(h))
    }
}