eson/structure
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

记录keydup

Browse Source
This commit is contained in:
huangsimin 2019-03-14 12:40:22 +08:00
parent 1a5b341e71
commit b5c4bd2fd0
2 changed files with 39 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
avlkey/avlkey.go
View File

@ -219,9 +219,9 @@ func (avl *AVL) GetNode(key interface{}) (*Node, bool) {
func (avl *AVL) Put(key, value interface{}) { func (avl *AVL) Put(key, value interface{}) {
avl.size++ avl.size++
node := &Node{key: key, value: value}
if avl.size == 1 { if avl.size == 1 {
avl.root = node avl.root = &Node{key: key, value: value}
return return
} }
@ -232,6 +232,7 @@ func (avl *AVL) Put(key, value interface{}) {
for { for {
if cur == nil { if cur == nil {
node := &Node{key: key, value: value}
parent.children[child] = node parent.children[child] = node
node.parent = parent node.parent = parent
node.child = child node.child = child
@ -243,14 +244,16 @@ func (avl *AVL) Put(key, value interface{}) {
} }
parent = cur parent = cur
c := avl.comparator(node.key, cur.key) c := avl.comparator(key, cur.key)
if c > -1 { // right child = (c + 2) / 2
child = 1 if c == 0 {
cur = cur.children[child] // node := &Node{key: key, value: value}
} else { cur.key = key
child = 0 cur.value = value
cur = cur.children[child] return
} }
cur = cur.children[child]
} }
} }
@ -320,7 +323,7 @@ func (avl *AVL) lrrotate(cur *Node) {
if mov.children[l] != nil { if mov.children[l] != nil {
movparent.children[r] = mov.children[l] movparent.children[r] = mov.children[l]
movparent.children[r].parent = movparent movparent.children[r].parent = movparent
movparent.children[r].child = 1 movparent.children[r].child = l
} else { } else {
movparent.children[r] = nil movparent.children[r] = nil
} }
@ -361,7 +364,7 @@ func (avl *AVL) rlrotate(cur *Node) {
if mov.children[l] != nil { if mov.children[l] != nil {
movparent.children[r] = mov.children[l] movparent.children[r] = mov.children[l]
movparent.children[r].parent = movparent movparent.children[r].parent = movparent
movparent.children[r].child = 1 movparent.children[r].child = r
} else { } else {
movparent.children[r] = nil movparent.children[r] = nil
} }
@ -398,11 +401,16 @@ func (avl *AVL) rrotate(cur *Node) {
mov.key, mov.value, cur.key, cur.value = cur.key, cur.value, mov.key, mov.value //交换值达到, 相对位移 mov.key, mov.value, cur.key, cur.value = cur.key, cur.value, mov.key, mov.value //交换值达到, 相对位移
if mov.children[l] != nil { // if mov.children[l] != nil {
mov.children[l].parent, cur.children[l] = cur, mov.children[l] // mov.children[l].parent = cur
} else { // cur.children[l] = mov.children[l]
cur.children[l] = nil // } else {
} // cur.children[l] = nil
// }
// mov.children[l] 不可能为nil
mov.children[l].parent = cur
cur.children[l] = mov.children[l]
if mov.children[r] != nil { if mov.children[r] != nil {
mov.children[l] = mov.children[r] mov.children[l] = mov.children[r]
@ -423,23 +431,27 @@ func (avl *AVL) rrotate(cur *Node) {
mov.height = getMaxChildrenHeight(mov) + 1 mov.height = getMaxChildrenHeight(mov) + 1
cur.height = getMaxChildrenHeight(cur) + 1 cur.height = getMaxChildrenHeight(cur) + 1
} }
func (avl *AVL) lrotate(cur *Node) { func (avl *AVL) lrotate(cur *Node) {
const l = 1 const l = 1
const r = 0 const r = 0
// 1 right 0 left
mov := cur.children[l] mov := cur.children[l]
mov.key, mov.value, cur.key, cur.value = cur.key, cur.value, mov.key, mov.value //交换值达到, 相对位移 mov.key, mov.value, cur.key, cur.value = cur.key, cur.value, mov.key, mov.value //交换值达到, 相对位移
if mov.children[l] != nil { // if mov.children[l] != nil {
mov.children[l].parent, cur.children[l] = cur, mov.children[l] // mov.children[l].parent = cur
} else { // cur.children[l] = mov.children[l]
cur.children[l] = nil // } else {
} // cur.children[l] = nil
// }
// 不可能为
mov.children[l].parent = cur
cur.children[l] = mov.children[l]
if mov.children[r] != nil { if mov.children[r] != nil {
mov.children[l] = mov.children[r] mov.children[l] = mov.children[r]
@ -553,15 +565,12 @@ func (avl *AVL) fixPutHeight(cur *Node) {
} else { } else {
avl.lrotate(cur) avl.lrotate(cur)
} }
} else if diff > 1 { } else if diff > 1 {
l := cur.children[0] l := cur.children[0]
if getHeight(l.children[1]) > getHeight(l.children[0]) { if getHeight(l.children[1]) > getHeight(l.children[0]) {
avl.rlrotate(cur) avl.rlrotate(cur)
} else { } else {
avl.rrotate(cur) avl.rrotate(cur)
} }
} else { } else {

6
avlkey/avlkey_test.go
View File

@ -128,7 +128,7 @@ func TestPutStable(t *testing.T) {
func TestPutComparatorRandom(t *testing.T) { func TestPutComparatorRandom(t *testing.T) {
for n := 0; n < 1000000; n++ { for n := 0; n < 400000; n++ {
avl := New(utils.IntComparator) avl := New(utils.IntComparator)
godsavl := avltree.NewWithIntComparator() godsavl := avltree.NewWithIntComparator()
@ -165,7 +165,9 @@ func TestPutComparatorRandom(t *testing.T) {
// avl := New(utils.IntComparator) // avl := New(utils.IntComparator)
// for i := 0; i < 15; i++ { // for i := 0; i < 15; i++ {
// avl.Put(randomdata.Number(0, 1000)) // avl.Put(randomdata.Number(0, 1000))
// } // } const l = 1
// const r = 0
// t.Error(avl.String()) // t.Error(avl.String())
// t.Error(avl.Get(500)) // t.Error(avl.Get(500))
// } // }