Golang 中的大顶堆或小顶堆

2024年2月5日 198次阅读 来源: FishTailStudio

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))
    }
}
    原文作者：FishTailStudio
    原文地址: https://blog.csdn.net/gibiguvuggu/article/details/124901197
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞