Удалить узел из дерева двоичного поиска

Question

Я работаю над деревом двоичного поиска впервые и пытаюсь удалить узел из дерева двоичного поиска.

Однако каждый раз, когда я удаляю его, а затем использую inOrder() или preOrder(), чтобы проверить, удалено оно или нет, оно все равно отображается, что означает, что мне не удалось удалить этот конкретный узел.

Это мой код:

public class BinaryTree<T extends Comparable<T>> {
    TreeNode<T> root;
    ArrayList<Integer> myArr = new ArrayList<Integer>();
    Random random;
}

public void add(T o) {
    if (root == null) {
        root = new TreeNode<T>(o);
    } else {
        root.insert(o);
    }
}

public TreeNode<T> deleteNode(TreeNode<T> root, int value) {
    if (root == null)
        return null;
    if (root.data > value) {
        root.left = deleteNode(root.left, value);
    } else if (root.data < value) {
        root.right = deleteNode(root.right, value);
    } else {
        // if nodeToBeDeleted have both children
        if (root.left != null && root.right != null) {
            TreeNode<T> temp = root;
            // Finding minimum element from right
            TreeNode<T> minNodeForRight = minimumElement(temp.right);
            // Replacing current node with minimum node from right subtree
            root.data = minNodeForRight.data;
            // Deleting minimum node from right now
            deleteNode(root.right, minNodeForRight.data);
        }
        // if nodeToBeDeleted has only left child
        else if (root.left != null) {
            root = root.left;
        }
        // if nodeToBeDeleted has only right child
        else if (root.right != null) {
            root = root.right;
        }
        // if nodeToBeDeleted do not have child (Leaf node)
        else
            root = null;
    }
    return root;
}

public TreeNode<T> minimumElement(TreeNode<T> root) {
    if (root.left == null){
        return root;
    } else {
        return minimumElement(root.left);
    }
}

public static void main(String[] args){
    BinaryTree<Integer> bst = new BinaryTree<Integer>();
    bst.add(20);
    bst.add(30);
    bst.add(40);
    bst.add(80);
    bst.add(1200);
    bst.inOrder();
    bst.deleteNode(bst.root, 80);
    bst.inOrder();
}

Как решить эту проблему?

Answer 1

Изменить следующую строку

// Deleting minimum node from right now
deleteNode(root.right, minNodeForRight.data);

На

// Deleting minimum node from right now
root.right = deleteNode(root.right, minNodeForRight.data);