添加了树的接口
This commit is contained in:
154
tree/heap/heap.go
Normal file
154
tree/heap/heap.go
Normal file
@@ -0,0 +1,154 @@
|
||||
package heap
|
||||
|
||||
import (
|
||||
"github.com/474420502/focus/compare"
|
||||
)
|
||||
|
||||
type Heap struct {
|
||||
size int
|
||||
elements []interface{}
|
||||
Compare compare.Compare
|
||||
}
|
||||
|
||||
func New(Compare compare.Compare) *Heap {
|
||||
h := &Heap{Compare: Compare}
|
||||
h.elements = make([]interface{}, 16, 16)
|
||||
return h
|
||||
}
|
||||
|
||||
func (h *Heap) Size() int {
|
||||
return h.size
|
||||
}
|
||||
|
||||
func (h *Heap) Values() []interface{} {
|
||||
return h.elements[0:h.size]
|
||||
}
|
||||
|
||||
func (h *Heap) grow() {
|
||||
ecap := len(h.elements)
|
||||
if h.size >= ecap {
|
||||
ecap = ecap << 1
|
||||
grow := make([]interface{}, ecap, ecap)
|
||||
copy(grow, h.elements)
|
||||
h.elements = grow
|
||||
}
|
||||
}
|
||||
|
||||
func (h *Heap) Empty() bool {
|
||||
return h.size < 1
|
||||
}
|
||||
|
||||
func (h *Heap) Clear() {
|
||||
h.size = 0
|
||||
}
|
||||
|
||||
func (h *Heap) Reborn() {
|
||||
h.size = 0
|
||||
h.elements = make([]interface{}, 16, 16)
|
||||
}
|
||||
|
||||
func (h *Heap) Top() (interface{}, bool) {
|
||||
if h.size != 0 {
|
||||
return h.elements[0], true
|
||||
}
|
||||
return nil, false
|
||||
}
|
||||
|
||||
func (h *Heap) Put(v interface{}) {
|
||||
if v == nil {
|
||||
return
|
||||
}
|
||||
|
||||
h.grow()
|
||||
|
||||
curidx := h.size
|
||||
h.size++
|
||||
// up
|
||||
for curidx != 0 {
|
||||
pidx := (curidx - 1) >> 1
|
||||
pvalue := h.elements[pidx]
|
||||
if h.Compare(v, pvalue) > 0 {
|
||||
h.elements[curidx] = pvalue
|
||||
curidx = pidx
|
||||
} else {
|
||||
break
|
||||
}
|
||||
}
|
||||
h.elements[curidx] = v
|
||||
}
|
||||
|
||||
func (h *Heap) slimming() {
|
||||
|
||||
elen := len(h.elements)
|
||||
if elen >= 32 {
|
||||
ecap := elen >> 1
|
||||
if h.size <= ecap {
|
||||
ecap = elen - (ecap >> 1)
|
||||
slimming := make([]interface{}, ecap, ecap)
|
||||
copy(slimming, h.elements)
|
||||
h.elements = slimming
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func (h *Heap) Pop() (interface{}, bool) {
|
||||
|
||||
if h.size == 0 {
|
||||
return nil, false
|
||||
}
|
||||
|
||||
curidx := 0
|
||||
top := h.elements[curidx]
|
||||
h.size--
|
||||
|
||||
h.slimming()
|
||||
|
||||
if h.size == 0 {
|
||||
return top, true
|
||||
}
|
||||
|
||||
downvalue := h.elements[h.size]
|
||||
var cidx, c1, c2 int
|
||||
var cvalue1, cvalue2, cvalue interface{}
|
||||
// down
|
||||
for {
|
||||
cidx = curidx << 1
|
||||
|
||||
c2 = cidx + 2
|
||||
if c2 < h.size {
|
||||
cvalue2 = h.elements[c2]
|
||||
|
||||
c1 = cidx + 1
|
||||
cvalue1 = h.elements[c1]
|
||||
|
||||
if h.Compare(cvalue1, cvalue2) >= 0 {
|
||||
cidx = c1
|
||||
cvalue = cvalue1
|
||||
} else {
|
||||
cidx = c2
|
||||
cvalue = cvalue2
|
||||
}
|
||||
} else {
|
||||
|
||||
c1 = cidx + 1
|
||||
if c1 < h.size {
|
||||
cvalue1 = h.elements[c1]
|
||||
cidx = c1
|
||||
cvalue = cvalue1
|
||||
} else {
|
||||
break
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
if h.Compare(cvalue, downvalue) > 0 {
|
||||
h.elements[curidx] = cvalue
|
||||
curidx = cidx
|
||||
} else {
|
||||
break
|
||||
}
|
||||
}
|
||||
h.elements[curidx] = downvalue
|
||||
return top, true
|
||||
}
|
||||
201
tree/heap/heap_test.go
Normal file
201
tree/heap/heap_test.go
Normal file
@@ -0,0 +1,201 @@
|
||||
package heap
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/474420502/focus/compare"
|
||||
)
|
||||
|
||||
func TestHeapPushTopPop(t *testing.T) {
|
||||
h := New(compare.Int)
|
||||
l := []int{9, 5, 15, 2, 3}
|
||||
ol := []int{15, 9, 5, 3, 2}
|
||||
for _, v := range l {
|
||||
h.Put(v)
|
||||
}
|
||||
|
||||
for _, tv := range ol {
|
||||
if v, isfound := h.Top(); isfound {
|
||||
if !(isfound && v == tv) {
|
||||
t.Error(v)
|
||||
}
|
||||
}
|
||||
|
||||
if v, isfound := h.Pop(); isfound {
|
||||
if !(isfound && v == tv) {
|
||||
t.Error(v)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if h.Size() != 0 {
|
||||
t.Error("heap size is not equals to zero")
|
||||
}
|
||||
}
|
||||
|
||||
// func Int(k1, k2 interface{}) int {
|
||||
// c1 := k1.(int)
|
||||
// c2 := k2.(int)
|
||||
// switch {
|
||||
// case c1 > c2:
|
||||
// return -1
|
||||
// case c1 < c2:
|
||||
// return 1
|
||||
// default:
|
||||
// return 0
|
||||
// }
|
||||
// }
|
||||
|
||||
// func TestPush(t *testing.T) {
|
||||
|
||||
// for i := 0; i < 1000000; i++ {
|
||||
// h := New(Int)
|
||||
|
||||
// gods := binaryheap.NewWithIntComparator()
|
||||
// for c := 0; c < 20; c++ {
|
||||
// v := randomdata.Number(0, 100)
|
||||
// h.Push(v)
|
||||
// gods.Push(v)
|
||||
// }
|
||||
|
||||
// r1 := spew.Sprint(h.Values())
|
||||
// r2 := spew.Sprint(gods.Values())
|
||||
// if r1 != r2 {
|
||||
// t.Error(r1)
|
||||
// t.Error(r2)
|
||||
// break
|
||||
// }
|
||||
// }
|
||||
|
||||
// }
|
||||
|
||||
// func TestPop(t *testing.T) {
|
||||
|
||||
// for i := 0; i < 200000; i++ {
|
||||
// h := New(Int)
|
||||
|
||||
// // m := make(map[int]int)
|
||||
// gods := binaryheap.NewWithIntComparator()
|
||||
// for c := 0; c < 40; c++ {
|
||||
// v := randomdata.Number(0, 100)
|
||||
// // if _, ok := m[v]; !ok {
|
||||
// h.Push(v)
|
||||
// gods.Push(v)
|
||||
// // m[v] = v
|
||||
// // }
|
||||
|
||||
// }
|
||||
|
||||
// // t.Error(h.Values())
|
||||
// // t.Error(gods.Values())
|
||||
// for c := 0; c < randomdata.Number(5, 10); c++ {
|
||||
// v1, _ := h.Pop()
|
||||
// v2, _ := gods.Pop()
|
||||
|
||||
// if v1 != v2 {
|
||||
// t.Error(h.Values(), v1)
|
||||
// t.Error(gods.Values(), v2)
|
||||
// return
|
||||
// }
|
||||
// }
|
||||
|
||||
// r1 := spew.Sprint(h.Values())
|
||||
// r2 := spew.Sprint(gods.Values())
|
||||
// if r1 != r2 {
|
||||
// t.Error(r1)
|
||||
// t.Error(r2)
|
||||
// break
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
|
||||
// func BenchmarkPush(b *testing.B) {
|
||||
|
||||
// l := loadTestData()
|
||||
|
||||
// b.ResetTimer()
|
||||
// execCount := 50
|
||||
// b.N = len(l) * execCount
|
||||
|
||||
// for c := 0; c < execCount; c++ {
|
||||
// b.StopTimer()
|
||||
// h := New(Int)
|
||||
// b.StartTimer()
|
||||
// for _, v := range l {
|
||||
// h.Push(v)
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
|
||||
// func BenchmarkPop(b *testing.B) {
|
||||
|
||||
// h := New(Int)
|
||||
|
||||
// l := loadTestData()
|
||||
|
||||
// b.ResetTimer()
|
||||
// execCount := 20
|
||||
// b.N = len(l) * execCount
|
||||
|
||||
// for c := 0; c < execCount; c++ {
|
||||
// b.StopTimer()
|
||||
// for _, v := range l {
|
||||
// h.Push(v)
|
||||
// }
|
||||
// b.StartTimer()
|
||||
// for h.size != 0 {
|
||||
// h.Pop()
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
|
||||
// func BenchmarkGodsPop(b *testing.B) {
|
||||
|
||||
// h := binaryheap.NewWithIntComparator()
|
||||
|
||||
// l := loadTestData()
|
||||
|
||||
// b.ResetTimer()
|
||||
// execCount := 20
|
||||
// b.N = len(l) * execCount
|
||||
|
||||
// for c := 0; c < execCount; c++ {
|
||||
// b.StopTimer()
|
||||
// for _, v := range l {
|
||||
// h.Push(v)
|
||||
// }
|
||||
// b.StartTimer()
|
||||
// for h.Size() != 0 {
|
||||
// h.Pop()
|
||||
// }
|
||||
// }
|
||||
|
||||
// }
|
||||
|
||||
// func BenchmarkGodsPush(b *testing.B) {
|
||||
// l := loadTestData()
|
||||
|
||||
// b.ResetTimer()
|
||||
// execCount := 50
|
||||
// b.N = len(l) * execCount
|
||||
|
||||
// for c := 0; c < execCount; c++ {
|
||||
// b.StopTimer()
|
||||
// h := binaryheap.NewWith(Int)
|
||||
// b.StartTimer()
|
||||
// for _, v := range l {
|
||||
// h.Push(v)
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
|
||||
// func loadTestData() []int {
|
||||
// data, err := ioutil.ReadFile("../l.log")
|
||||
// if err != nil {
|
||||
// log.Println(err)
|
||||
// }
|
||||
// var l []int
|
||||
// decoder := gob.NewDecoder(bytes.NewReader(data))
|
||||
// decoder.Decode(&l)
|
||||
// return l
|
||||
// }
|
||||
Reference in New Issue
Block a user