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