mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2025-04-06 05:45:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse Source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2024-10-12 14:26:30 +00:00
parent 0719a18d47
commit ca0ee11eb5
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
data_structures/binary_tree/avl_tree.py
View File

@ -199,10 +199,10 @@ def del_node(root: MyNode, data: Any) -> MyNode | None:
if root is None:
print("Nothing to delete")
return None
left_child = root.get_left()
right_child = root.get_right()
if root.get_data() == data:
if left_child is not None and right_child is not None:
temp_data = get_left_most(right_child)
@ -226,10 +226,12 @@ def del_node(root: MyNode, data: Any) -> MyNode | None:
else:
root.set_right(del_node(right_child, data))
root.set_height(my_max(get_height(root.get_right()), get_height(root.get_left())) + 1)
root.set_height(
my_max(get_height(root.get_right()), get_height(root.get_left())) + 1
)
balance_factor = get_height(root.get_left()) - get_height(root.get_right())
if balance_factor == 2:
assert right_child is not None
if get_height(right_child.get_right()) > get_height(right_child.get_left()):
@ -242,7 +244,7 @@ def del_node(root: MyNode, data: Any) -> MyNode | None:
root = right_rotation(root)
else:
root = lr_rotation(root)
return root