wavl/python
1
0
Fork 0
Code Issues Pull requests Projects 1 Releases 1 Wiki Activity

chore(avl): refactor

Browse source 
Fixes #2

Signed-off-by: Matej Focko <mfocko@redhat.com>
This commit is contained in:
Matej Focko 2022-05-12 12:20:55 +02:00
parent 705d5cfa17
commit 88d691225e
Signed by: mfocko
GPG key ID: 7C47D46246790496
1 changed files with 12 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
avl.py
View file

@ -124,20 +124,17 @@ class AVLTree(RankedTree[T]):
if rotating_around_root:
self.root = new_root
return factor == 0
return factor != 0
def __delete_fixup(
self, y: Optional[Node[T]], parent: Optional[Node[T]] = None
self, x: Optional[Node[T]], parent: Optional[Node[T]] = None
) -> bool:
x = y if y else parent
assert x
factor = _balance_factor(x)
if factor == 0:
_update_rank(x)
return False
elif factor in (-1, 1):
return True
elif factor in (-1, 1):
return False
rotating_around_root = x.parent is None
y, leaning, to_left, to_right = (
@ -158,11 +155,13 @@ class AVLTree(RankedTree[T]):
def _delete_rebalance(
self, node: Optional[Node[T]], parent: Optional[Node[T]]
) -> None:
while node or parent:
# TODO: Check if it is possible to not propagate all the way up.
# if self.__delete_fixup(node, parent):
# return
self.__delete_fixup(node, parent)
node, parent = parent, (parent.parent if parent else None)
if not node and not parent:
return
if not node:
node, parent = parent, parent.parent
while node and self.__delete_fixup(node, parent):
node, parent = parent, parent.parent if parent else None
# endregion DeleteRebalance