Skip to content

Commit

Permalink
fix(heap): fix go code
Browse files Browse the repository at this point in the history
  • Loading branch information
Reanon committed Jan 13, 2023
1 parent 32962fb commit 79d51e3
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions codes/go/chapter_heap/heap.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,9 +4,9 @@

package chapter_heap

// intHeap 是一个由整数组成的最大堆
// intHeap 是一个由整数组成的堆
// 通过实现 heap.Interface 来构建堆
type intHeap []any
type intHeap []int

// Len sort.Interface 的方法
func (h *intHeap) Len() int {
Expand All @@ -16,7 +16,7 @@ func (h *intHeap) Len() int {
// Less sort.Interface 的方法
func (h *intHeap) Less(i, j int) bool {
// 如果实现小顶堆,则需要调整为小于号
return (*h)[i].(int) > (*h)[j].(int)
return (*h)[i] > (*h)[j]
}

// Swap sort.Interface 的方法
Expand All @@ -34,11 +34,11 @@ func (h *intHeap) Push(x any) {

// Top 获取堆顶元素
func (h *intHeap) Top() int {
return (*h)[0].(int)
return (*h)[0]
}

// Pop heap.Interface 的方法,实现弹出堆顶元素
func (h *intHeap) Pop() any {
func (h *intHeap) Pop() int {
// 待出堆元素存放在最后
last := (*h)[len(*h)-1]
*h = (*h)[:len(*h)-1]
Expand Down

0 comments on commit 79d51e3

Please sign in to comment.