LeetCode每日打卡 450. 删除二叉搜索树中的节点

题目介绍

给定一个二叉搜索树的根节点 root 和一个值 key，删除二叉搜索树中的 key 对应的节点，并保证二叉搜索树的性质不变。返回二叉搜索树（有可能被更新）的根节点的引用。

一般来说，删除节点可分为两个步骤：

1、首先找到需要删除的节点；
2、如果找到了，删除它。

输入：root = [5,3,6,2,4,null,7], key = 3
输出：[5,4,6,2,null,null,7]
解释：给定需要删除的节点值是 3，所以我们首先找到 3 这个节点，然后删除它。
一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。
另一个正确答案是 [5,2,6,null,4,null,7]。

 

解题思路

递归，迭代

    public static void main(String[] args) {
        TreeNode root = new StrToTreeNode().createTree("5,3,6,2,4,null,7");
        TreeNode treeNode = deleteNode(root, 3);
        System.out.println(treeNode);
    }

    public static TreeNode deleteNode(TreeNode root, int key) {
        // root为空，代表为搜索值为key的节点，awa当然也不可能搜索不到
        if (root == null) {
            return null;
        }
        // key节点可能存于左子树中，去左子树中去删除
        else if (key < root.val) {
            root.left = deleteNode(root.left, key);
        }
        // key节点可能存于右子树中，去右子树中删除
        else if (key > root.val) {
            root.right = deleteNode(root.right, key);
        }
        // 当前节点就是要删除的节点
        else {
            // 删除的节点没有左子树
            if (root.left == null) {
                return root.right;
            }
            // 删除的节点没有右子树
            else if (root.right == null) {
                return root.left;
            }
            // 删除的节点左右子树都有
            else if (root.left != null && root.right != null) {
                TreeNode rightTree = root.right;
                // 查找删除节点右子树中的最左节点
                while (rightTree.left != null) {
                    rightTree = rightTree.left;
                }
                // 将删除节点的左子树替换为右子树的最左节点的左子树
                rightTree.left = root.left;
                // 删除系欸但的右子树顶替其位置，节点被删除
                root = root.right;
            } else {//如果不存在右侧最小的叶子节点，root的右子树为空，直接返回左子树
                return root.left;
            }
        }
        return root;
    }