刷刷刷 Day 22 | 450. 删除二叉搜索树中的节点

450. 删除二叉搜索树中的节点

LeetCode题目要求

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

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

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

示例

输入：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]。

解题思路

由于是删除节点操作，删除后还需要保证是一个二叉搜索树。

1. 如果删除的是叶子节点，那么不需要又任何变化，直接删除即可
2. 如果是非叶子节点，存在左右子树，将右子树升级为父结点
3. 如果是非叶子节点，只存在左子树，那么左子树升级为父结点
4. 如果是非叶子节点，只存在右子树，那么右子树升级为父节点

上代码

class Solution {
    public TreeNode deleteNode(TreeNode root, int key) {
        /**
        1. 如果删除的是叶子节点，那么不需要又任何变化，直接删除即可
        2. 如果是非叶子节点，存在左右子树，将右子树升级为父结点
        3. 如果是非叶子节点，只存在左子树，那么左子树升级为父结点
        4. 如果是非叶子节点，只存在右子树，那么右子树升级为父节点
        */
 
        if (root == null) {
            return root;
        }
 
        if (root.val == key) {
            if (root.left == null && root.right == null) {
                return null;
            } else if (root.left == null) {
                return root.right;
            } else if (root.right == null) {
                return root.left;
            } else {
                TreeNode node = root.right;
                while (node.left != null) {
                    node = node.left;
                }
                node.left = root.left;
                root = root.right;
                return root;
            }
        }
 
        //  往左子树遍历
        if (root.val > key) {
            root.left = deleteNode(root.left, key);
        }
        // 往右子树遍历
        if (root.val < key) {
            root.right = deleteNode(root.right, key);
        }
        return root;
    }
}

重难点

搞明白以下集中情况， 并结合二叉搜索树的特点

1. 如果删除的是叶子节点，那么不需要又任何变化，直接删除即可
2. 如果是非叶子节点，存在左右子树，将右子树升级为父结点
3. 如果是非叶子节点，只存在左子树，那么左子树升级为父结点
4. 如果是非叶子节点，只存在右子树，那么右子树升级为父节点
   附：学习资料链接