Leetcode 450:删除二叉搜索树中的节点

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

一般来说,删除节点可分为两个步骤:

  1. 首先找到需要删除的节点;
  2. 如果找到了,删除它。
public TreeNode deleteNode(TreeNode root, int key) {
        if (root == null) return root;

        //找到了删除结点
        if (root.val == key) {
            //1.该结点左子树为空,右子树直接补上
            if (root.left == null) {
                return root.right;
            } else if (root.right == null) {   //2.右子树为空,左子树补上
                return root.left;
            } else {
                //3.左右子树都不为空
                TreeNode newNode = root.right;  
                while (newNode.left != null) {    //找到右子树的最左边结点
                    newNode = newNode.left;
                }
                newNode.left = root.left;    //将要删除节点的左子树放在上述右子树的最左边结点上
                root = root.right;   //用右子树补上删除结点的位置
                return root;
            }
        }
        if (root.val > key) root.left = deleteNode(root.left, key);
        if (root.val < key) root.right = deleteNode(root.right, key);
        return root;
    }

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值