fuhouyin/hello-algo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(heap): fix go code

Browse Source
This commit is contained in:
reanon 2023-01-12 20:58:39 +08:00
parent 79d51e3691
commit 8117a1d47d
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
codes/go/chapter_heap/heap.go
View File

@ -6,7 +6,7 @@ package chapter_heap
// intHeap 是一个由整数组成的堆 // intHeap 是一个由整数组成的堆
// 通过实现 heap.Interface 来构建堆 // 通过实现 heap.Interface 来构建堆
type intHeap []int type intHeap []any
// Len sort.Interface 的方法 // Len sort.Interface 的方法
func (h *intHeap) Len() int { func (h *intHeap) Len() int {
@ -16,7 +16,7 @@ func (h *intHeap) Len() int {
// Less sort.Interface 的方法 // Less sort.Interface 的方法
func (h *intHeap) Less(i, j int) bool { func (h *intHeap) Less(i, j int) bool {
// 如果实现小顶堆，则需要调整为小于号 // 如果实现小顶堆，则需要调整为小于号
return (*h)[i] > (*h)[j] return (*h)[i].(int) > (*h)[j].(int)
} }
// Swap sort.Interface 的方法 // Swap sort.Interface 的方法
@ -33,12 +33,12 @@ func (h *intHeap) Push(x any) {
} }
// Top 获取堆顶元素 // Top 获取堆顶元素
func (h *intHeap) Top() int { func (h *intHeap) Top() any {
return (*h)[0] return (*h)[0]
} }
// Pop heap.Interface 的方法，实现弹出堆顶元素 // Pop heap.Interface 的方法，实现弹出堆顶元素
func (h *intHeap) Pop() int { func (h *intHeap) Pop() any {
// 待出堆元素存放在最后 // 待出堆元素存放在最后
last := (*h)[len(*h)-1] last := (*h)[len(*h)-1]
*h = (*h)[:len(*h)-1] *h = (*h)[:len(*h)-1]