力扣450_删除二叉搜索树中的节点

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

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

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

示例：

算法思想：
根据递归的思想，我们很容易写出如下代码

TreeNode deleteNode(TreeNode root, int key) {
    if (root.val == key) {
        // 找到啦，进行删除
    } else if (root.val > key) {
        // 去左子树找
        root.left = deleteNode(root.left, key);
    } else if (root.val < key) {
        // 去右子树找
        root.right = deleteNode(root.right, key);
    }
    return root;
}

问题就在于找到了待删除的节点后，怎么去删除的问题，因为我们不能破坏二叉搜索树的特性，我们可以分类讨论。

1.待删除的节点恰好是末端顶点，直接删除

if (root.left == null && root.right == null)
    return null;

2.待删除的节点只有一个非空节点，子节点直接替换根

if (root.left == null) return root.right;
if (root.right == null) return root.left;

3.待删除的节点有两个子节点，必须找到左子树最大的节点，或者右子树最小的节点来替换根，这里我们找左子树最大的节点

if(root.left != null && root.right != null) {
	// 找到左子树的最大节点
    TreeNode maxNode= getLeftMax(root.left);
    // 把 root 改成 maxNode
    root.val = maxNode.val;
    // 转而去删除 maxNode
    root.left = deleteNode(root.right, maxNode.val);
}

框架已经搭好，代码就呼之欲出了

算法实现：

public TreeNode deleteNode(TreeNode root, int key) {

        if(root == null) return null;

        if(root.val > key) {
            root.left = deleteNode(root.left, key);
        }

        if(root.val < key) {
            root.right = deleteNode(root.right, key);
        }

        if(root.val == key) {

            //1.如果左右为空 2.如果左右有一个为空
            if(root.left == null) return root.right;
            if(root.right == null) return root.left;

            TreeNode maxNode = getLeftMax(root.left); //找到左边最大的节点
            root.val = maxNode.val; //替换掉根节点的值

            root.left = deleteNode(root.left, maxNode.val); //删除左边最大节点
        }

        return root;

    }

    public TreeNode getLeftMax(TreeNode root) {
        while(root.right!=null) {
            root = root.right;
        }
        return root;
    }