Update bst_delete_node.py

This commit is contained in:
bt3gl 2023-08-08 15:41:34 -07:00 committed by GitHub
parent 122a535241
commit a3ce5db3a5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -21,14 +21,14 @@ def predecessor(root):
def delete_node(root, key):
if not root:
if root is None:
return root
if key > root.val:
root.right = deleteNode(root.right, key)
root.right = delete_node(root.right, key)
elif key < root.val:
root.left = deleteNode(root.left, key)
root.left = delete_node(root.left, key)
else:
if not (root.left or root.right):
@ -36,11 +36,11 @@ def delete_node(root, key):
elif root.right:
root.val = successor(root)
root.right = deleteNode(root.right, root.val)
root.right = delete_node(root.right, root.val)
else:
root.val = predecessor(root)
root.left = deleteNode(root.left, root.val)
root.left = delete_node(root.left, root.val)
return root