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

题目链接
思路：递归
分析：删除root中值为key的节点，首先分为两步：
第一步：找到值为key的节点
第二步：删除值为key的节点
完成第一步的方案：
因为树为二叉搜索树，所以有左小右大的原则，可以在O(h)的时间复杂度内找到值为key的节点。（h为数的高度）
具体做法是：

1. root.val>key，代表值为key的节点在root的左子数里面，那么就递归从root的左子树中删除值位key的节点。

2. root.val<key，代表值为key的节点在root的右子树里面，那么就递归从root的右子树中删除值为key的节点。

3. 如果root为null，说明没有找到值为key的节点，按照题目意思，不用删除任何东西，那么就直接返回null。

4. 如果root.val==key，那么说明完成了第一步，找到了值为key的节点。此时要分四种情况。
   4.1 此时root为叶子节点：即root.left=null并且root.right=null。那么就直接删除当前root即可，即直接返回null。
   4.2 此时root节点只有左孩子，那么也是删除当前root节点，但是要返回root的左孩子。
   4.3 此时root节点只有右孩子，那么也是删除当前root节点，但是要返回root的右孩子。
   4.4 此时root既有左孩子也有右孩子。这里又有两种做法。

   1. 删除当前节点，从当前节点的左子树中选出最大的节点来代替当前节点。
   2. 删除当前节点，从当前节点的右子树中选出最大的节点来代替当前节点。

这里我就采用了第二种方法，从当前节点的右子树中选出最大的节点来代替当前节点。

代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode deleteNode(TreeNode root, int key) {
        if(root==null){
            //表示没有找到key
            return null;
        }
        if(root.val>key){
            //key在左子树中，那么递归从左子树中删除值为key的节点
            //那么root.left就是递归后的返回的根节点
            root.left = deleteNode(root.left, key);
            return root;
        }else if(root.val<key){
            //同上
            root.right = deleteNode(root.right, key);
            return root;
        }else{
            //root.val==key
            
            //key在叶子节点上
            if(root.left==null && root.right==null){
                return null;
            }
            //如果只有一个孩子
            if(root.right==null){
                //只有左孩子，那么直接返回左孩子，这样做就是删除了root自身，好好想想
                return root.left;
            }
            if(root.left==null){
                //同上
                return root.right;
            }

            //此时说明左右孩子都不为null
            //采用第二种做法，从右孩子中找出最小的节点
            TreeNode successor = root.right;
            while(successor.left!=null){
                successor = successor.left;
            }

            //此时successor就是root的右孩子中值最小的节点
            //从root中删除子节点 successor
            root.right = deleteNode(root.right, successor.val);

            //此时使用successor也就是右孩子中值最小的节点代替当前root节点
            successor.left = root.left;
            successor.right = root.right;
            //返回此时的根节点
            return successor;
        }
    }
}