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

improved del_node func(11)

Browse Source
This commit is contained in:
Shaurya Bisht 2024-10-13 11:14:23 +00:00
parent ceeb5b7ab2
commit 38ea09c586
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
data_structures/binary_tree/avl_tree.py
View File

@ -209,7 +209,6 @@ def get_min_value_node(node: MyNode) -> MyNode:
current_node = current_node.get_left() current_node = current_node.get_left()
return current_node return current_node
def del_node(root: MyNode | None, data: Any) -> MyNode | None: def del_node(root: MyNode | None, data: Any) -> MyNode | None:
if root is None: if root is None:
print("Nothing to delete") print("Nothing to delete")
@ -224,11 +223,10 @@ def del_node(root: MyNode | None, data: Any) -> MyNode | None:
return root.get_right() return root.get_right()
elif root.get_right() is None: elif root.get_right() is None:
return root.get_left() return root.get_left()
right_child = root.get_right() right_child = root.get_right()
assert right_child is not None # Ensure root.get_right() is not None assert right_child is not None
temp = get_min_value_node(right_child) temp = get_min_value_node(right_child)
assert temp is not None # Ensure get_min_value_node doesn't return None
root.set_data(temp.get_data()) root.set_data(temp.get_data())
root.set_right(del_node(root.get_right(), temp.get_data())) root.set_right(del_node(root.get_right(), temp.get_data()))