From 45ef1aca6cbb2e7eb132905ef5770bfe5946d580 Mon Sep 17 00:00:00 2001 From: huangsimin Date: Thu, 28 Mar 2019 18:14:15 +0800 Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E5=96=84=E4=BA=86rotate=20=E8=BF=98?= =?UTF-8?q?=E5=B7=AE3=E7=9A=84=E7=89=B9=E6=AE=8A=E6=97=8B=E8=BD=AC?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- priority_queue/vbt.go | 20 +++++++++++++++----- 1 file changed, 15 insertions(+), 5 deletions(-) diff --git a/priority_queue/vbt.go b/priority_queue/vbt.go index 9f7c61b..393f90b 100644 --- a/priority_queue/vbt.go +++ b/priority_queue/vbt.go @@ -440,7 +440,7 @@ func (tree *vbTree) Put(key interface{}) *tNode { ls, rs := cur.children[0].size, cur.children[1].size factor := cur.size / 10 // or factor = 1 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 { cur.parent.children[r] = lrn } + } else { + tree.root = lrn } + lrn.parent = cur.parent lrn.children[l] = cur.children[l] lrn.children[l].parent = lrn - lrn.children[l] = cur - lrn.children[l].parent = lrn + lrn.children[r] = cur + lrn.children[r].parent = lrn cur.children[l] = lrrn cur.children[l].parent = cur @@ -701,13 +704,16 @@ func (tree *vbTree) rlrotate(cur *tNode) *tNode { } else { cur.parent.children[r] = lrn } + } else { + tree.root = lrn } + lrn.parent = cur.parent lrn.children[l] = cur.children[l] lrn.children[l].parent = lrn - lrn.children[l] = cur - lrn.children[l].parent = lrn + lrn.children[r] = cur + lrn.children[r].parent = lrn cur.children[l] = lrrn cur.children[l].parent = cur @@ -752,6 +758,8 @@ func (tree *vbTree) rrotate(cur *tNode) *tNode { } else { cur.parent.children[r] = ln } + } else { + tree.root = ln } ln.parent = cur.parent @@ -802,6 +810,8 @@ func (tree *vbTree) lrotate(cur *tNode) *tNode { } else { cur.parent.children[r] = ln } + } else { + tree.root = ln } ln.parent = cur.parent