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

完善了rotate 还差3的特殊旋转

Browse Source
This commit is contained in:
huangsimin 2019-03-28 18:14:15 +08:00
parent 6f6a64ed08
commit 45ef1aca6c
1 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
priority_queue/vbt.go
View File

@ -440,7 +440,7 @@ func (tree *vbTree) Put(key interface{}) *tNode {
ls, rs := cur.children[0].size, cur.children[1].size ls, rs := cur.children[0].size, cur.children[1].size
factor := cur.size / 10 // or factor = 1 factor := cur.size / 10 // or factor = 1
if rs >= ls*2+factor || ls >= rs*2+factor { if rs >= ls*2+factor || ls >= rs*2+factor {
tree.fixSize(cur, ls, rs) cur = tree.fixSize(cur, ls, rs)
} }
} }
@ -640,13 +640,16 @@ func (tree *vbTree) lrrotate(cur *tNode) *tNode {
} else { } else {
cur.parent.children[r] = lrn cur.parent.children[r] = lrn
} }
} else {
tree.root = lrn
} }
lrn.parent = cur.parent
lrn.children[l] = cur.children[l] lrn.children[l] = cur.children[l]
lrn.children[l].parent = lrn lrn.children[l].parent = lrn
lrn.children[l] = cur lrn.children[r] = cur
lrn.children[l].parent = lrn lrn.children[r].parent = lrn
cur.children[l] = lrrn cur.children[l] = lrrn
cur.children[l].parent = cur cur.children[l].parent = cur
@ -701,13 +704,16 @@ func (tree *vbTree) rlrotate(cur *tNode) *tNode {
} else { } else {
cur.parent.children[r] = lrn cur.parent.children[r] = lrn
} }
} else {
tree.root = lrn
} }
lrn.parent = cur.parent
lrn.children[l] = cur.children[l] lrn.children[l] = cur.children[l]
lrn.children[l].parent = lrn lrn.children[l].parent = lrn
lrn.children[l] = cur lrn.children[r] = cur
lrn.children[l].parent = lrn lrn.children[r].parent = lrn
cur.children[l] = lrrn cur.children[l] = lrrn
cur.children[l].parent = cur cur.children[l].parent = cur
@ -752,6 +758,8 @@ func (tree *vbTree) rrotate(cur *tNode) *tNode {
} else { } else {
cur.parent.children[r] = ln cur.parent.children[r] = ln
} }
} else {
tree.root = ln
} }
ln.parent = cur.parent ln.parent = cur.parent
@ -802,6 +810,8 @@ func (tree *vbTree) lrotate(cur *tNode) *tNode {
} else { } else {
cur.parent.children[r] = ln cur.parent.children[r] = ln
} }
} else {
tree.root = ln
} }
ln.parent = cur.parent ln.parent = cur.parent