mirror of
https://github.com/halfrost/LeetCode-Go.git
synced 2025-07-04 16:12:47 +08:00
31 lines
512 B
Go
31 lines
512 B
Go
package structures
|
||
|
||
import (
|
||
"container/heap"
|
||
"fmt"
|
||
"testing"
|
||
|
||
"github.com/stretchr/testify/assert"
|
||
)
|
||
|
||
func Test_intHeap(t *testing.T) {
|
||
ast := assert.New(t)
|
||
|
||
ih := new(intHeap)
|
||
heap.Init(ih)
|
||
|
||
heap.Push(ih, 1)
|
||
heap.Pop(ih)
|
||
|
||
begin, end := 0, 10
|
||
for i := begin; i < end; i++ {
|
||
heap.Push(ih, i)
|
||
ast.Equal(0, (*ih)[0], "插入 %d 后的最小值却是 %d，ih=%v", i, (*ih)[0], (*ih))
|
||
}
|
||
|
||
for i := begin; i < end; i++ {
|
||
fmt.Println(i, *ih)
|
||
ast.Equal(i, heap.Pop(ih), "Pop 后 ih=%v", (*ih))
|
||
}
|
||
}
|