删除二叉搜索树中的节点——leetcode450

删除二叉搜索树中的节点

题目：删除二叉搜索树中的节点

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

示例 1:

输入：root = [5,3,6,2,4,null,7], key = 3
输出：[5,4,6,2,null,null,7]

题解

1.删除的节点为叶子节点，则直接删除此节点

2.删除的节点有右(左)孩子，则当前节点等于右(左)节点。

3.删除的节点有左右孩子，找到右子树的最小节点或左子树的最大节点来代替要删除的节点

class Solution {
    public TreeNode findAndDelete(TreeNode root, int key)
    {
        if(root==null) return null;
        if(root.val==key) {
            if(root.left==null && root.right==null) return null;
            if(root.left==null) return root.right;
            if(root.right==null) return root.left;
            TreeNode minNode=findMinNode(root.right);
            root.val=minNode.val;
            root.right= findAndDelete(root.right, minNode.val);
        }
        if(root.val>key) root.left= findAndDelete(root.left, key);
        else root.right= findAndDelete(root.right, key);
        return root;
    }

    public TreeNode findMinNode(TreeNode node)
    {
        if(node.left==null) return node;
        return findMinNode(node.left);
    }
    public TreeNode deleteNode(TreeNode root, int key) {
        return findAndDelete(root, key);
    }
}