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

给定一个二叉树和一个key，删除二叉树中值为key的节点。

删除节点还要保持BST的特性，那么有三种情况：
1、该节点只有左子树
2、该节点只有右子树
3、该节点既有左子树也有右子树
前两种情况比较简单，就直接拿左子树的根或者右子树的根替换就可以了
第三种情况，我们有两种选择，一是我们可以选取左子树的最大值来替换，二是我们可以选取右子树的最小值来替换

class Solution {
    public TreeNode deleteNode(TreeNode root, int key) {
        if(root == null)return root;
        if (root.val == key){
        	//找到节点
            if(root.left == null)return root.right;
            if (root.right == null)return root.left;
            //将右子树中最小的节点的值进行交换
            TreeNode minNode = getMinNode(root.right);
            root.val = minNode.val;
            //将问题转换为删除minNode
            root.right = deleteNode(root.right,minNode.val);
        }

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

    private TreeNode getMinNode(TreeNode node) {
        if (node.left == null)return node;
        return getMinNode(node.left);
    }
}