How do I delete a node from a binary tree in Python?

How do I delete a node from a binary tree in Python?

To delete a node from a binary tree in Python, you need to consider three cases: deleting a leaf node, deleting a node with one child, and deleting a node with two children. Which case are you dealing with?

I'm trying to delete a node with two children.

In that case, you need to find the node's in-order predecessor or successor. Replace the node's value with that of the predecessor/successor, then delete the predecessor/successor node. Would you like a code example?

Yes, please provide a code example.

Here's a simple implementation:

class Node:
    def __init__(self, key):
        self.left = None
        self.right = None
        self.val = key

def delete_node(root, key):
    if root is None:
        return root
    if key < root.val:
        root.left = delete_node(root.left, key)
    elif key > root.val:
        root.right = delete_node(root.right, key)
    else:
        if root.left is None:
            return root.right
        elif root.right is None:
            return root.left
        temp = min_value_node(root.right)
        root.val = temp.val
        root.right = delete_node(root.right, temp.val)
    return root

# Function to find the node with the minimum value

def min_value_node(node):
    current = node
    while current.left is not None:
        current = current.left
    return current

This code handles the deletion of a node with two children.

Thank you! This is really helpful.