力扣:删除二叉搜索树中的节点java

力扣:删除二叉搜索树中的节点java

在这里插入图片描述
*主要问题:删除节点有左孩子或右孩子,怎么办?
对于没有左右孩子的,直接返回null,即删除了当前节点
有一个孩子的,返回另一个孩子。
左右孩子都有的,将左孩子赋值为右孩子的最左节点的左子树。

流程:

递归三部曲
参数:返回根节点,参数为根节点和目标值
结束条件:节点为空就返回null(情况一,未找到)
单层递归逻辑
找到节点,就对剩余的四种情况判断
/情况二,找到了,该节点的左右孩子都为空
/情况三,找到了,该节点左孩子空,右孩子不空,则返回右孩子
/情况四,找到了,该节点右孩子空,左孩子不空,则返回左孩子
/情况五,找到了,左右孩子都不为空,则先设置一个指针,使其指向右孩子的最左叶子结点,然后将该最左节叶子节点赋值为当前目标节点的左孩子。
判断当前节点大于key 就从新赋值左节点为递归调用函数左子树为参数
判断当前节点小于key 就从新赋值右节点为递归调用函数右子树为参数
返回根节点
代码:

class Solution {
    public TreeNode deleteNode(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&&root.right !=null) return root.right;//情况三,找到了,该节点左孩子空,右孩子不空,则返回右孩子
            if(root.left != null&&root.right ==null) return root.left;//情况四,找到了,该节点右孩子空,左孩子不空,则返回左孩子
            if(root.left != null&&root.right !=null){//情况五,找到了,左右孩子都不为空,则先设置一个指针,使其指向右孩子的最左叶子结点,然后将该最左节叶子节点赋值为当前目标节点的左孩子。
                TreeNode cur = root.right;
                while(cur.left !=null){//找到右孩子的最左叶子节点
                    cur = cur.left;
                }
                cur.left = root.left;
                
                return root.right;//返回节点右节点,由上一层删除目标节点
            }
        }
        if(key>root.val) root.right = deleteNode(root.right,key);//通过判断key的范围,判断向左还是向右遍历
        if(key<root.val) root.left = deleteNode(root.left,key);
        return root;//返回根节点

        
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值