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

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

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

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

示例 1:

img

1
2
3
4
输入：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], 如下图所示。

Solution

二叉树通用的删除方法

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
public TreeNode deleteNode(TreeNode root, int key) {
    if (root == null) return null; // 如果树为空或达到空节点，返回null

    if (root.val == key) {
        // 要删除的节点找到了
        if (root.right == null) {
            return root.left; // 如果无右子树，返回左子树
        }
        // 找到右子树的最小节点，即右子树最左边的节点
        TreeNode cur = root.right;
        while (cur.left != null) {
            cur = cur.left;
        }
        // 交换当前节点和右子树最小节点的值
        root.val = cur.val;
        cur.val = key;
        // 删除右子树中的key节点
        root.right = deleteNode(root.right, key);
    } else if (key < root.val) {
        // 递归左子树
        root.left = deleteNode(root.left, key);
    } else {
        // 递归右子树
        root.right = deleteNode(root.right, key);
    }
    return root;
}