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

删除节点可以分三种情况:

  1. 被删除节点为叶子节点,此时只需要直接删除即可;
  2. 被删除节点含有一个子节点,此时可以将该节点的子节点替换掉该节点,即将子节点的引用赋值给被删除节点的父节点;
  3. 被删除节点含有左子树和右子树,此时可以利用左子树的最大节点或右子树的最小节点来替换掉被删除节点。
class Solution {
    public TreeNode deleteNode(TreeNode root, int key) {
        if(root == null)return null;
        else if(key<root.val){
            root.left = deleteNode(root.left, key);
            return root;
        }else if(key>root.val){
            root.right = deleteNode(root.right, key);
            return root;
        }else{
            //找到被删除节点
            if(root.left == null){
                return root.right;
            }else{
                if(root.right == null){
                    return root.left;
                }else{
                    TreeNode n = findMin(root.right);
                    n.right = deleteNode(root.right, n.val);
                    n.left = root.left;
                    return n;
                }
            }
        }
    }
    public TreeNode findMin(TreeNode node){
        if(node.left==null)return node;
        return findMin(node.left);   
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值