其他分享
首页 > 其他分享> > 删除二叉搜索树中的节点——leetcode450

删除二叉搜索树中的节点——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);
    }
}

 

标签:right,return,leetcode450,树中,二叉,key,null,root,节点
来源: https://www.cnblogs.com/greengages/p/15762034.html