450. 删除二叉搜索树中的节点 701. 二叉搜索树中的插入操作

        前驱结点: 先取当前节点的左节点,取该节点的最右节点
        后继结点: 先取当前节点的右节点,取该节点的最左节点 

    算法:

        如果 key > root.val,说明要删除的节点在右子树,root.right = deleteNode(root.right, key)。
        如果 key < root.val,说明要删除的节点在左子树,root.left = deleteNode(root.left, key)。
        如果 key == root.val,则该节点就是我们要删除的节点,则:
        ①:如果该节点是叶子节点,则直接删除它:root = null。
        ②:如果该节点不是叶子节点且有右节点,则用它的后继节点的值替代 root.val = successor.val,然后删除后继节点。
        ③:如果该节点不是叶子节点且只有左节点,则用它的前驱节点的值替代 root.val = predecessor.val,然后删除前驱节点
        (为什么不直接返回root.left,因为这时候的root可能是子树,所以连接到root.left,
        但是不是返回root.left,而是返回根节点)
        ④:返回 root。
class Solution {
    public TreeNode deleteNode(TreeNode root, int key) {
        if(root==null){
            return null;
        }
        //左右子树递归
        if(root.val<key){
            root.right=deleteNode(root.right,key);
        }
        if(root.val>key){
            root.left=deleteNode(root.left,key);
        }
        //root是要找的那个值
        if(root.val==key){
        	//是叶子节点
            if(root.left==null&&root.right==null){
                root=null;
            }
            // 要删除的节点不是叶子节点,并且有右子树
            else if(root.right!=null){
                root.val=laternode(root);
                root.right=deleteNode(root.right,root.val);
            }
            //要删除的节点不是叶子节点,只有左子树 没有右子树
            else{
                root.val=prenode(root);
                root.left=deleteNode(root.left,root.val);
                //现在已经把前驱节点的值赋给了root,即现在root和pre的值均为pre->val,
                //所以现在应该删除pre节点,即左子树中数值为root->val(pre->val)的节点
            }      
        }
        return root;

    }
    //找前驱节点
    int prenode(TreeNode root){
        TreeNode node=root.left;
        while(node.right!=null){
            node=node.right;
        }
        return node.val;

    }
    //找后继节点
    int laternode(TreeNode root){
        TreeNode node=root.right;
        while(node.left!=null){
            node=node.left;
        }
        return node.val;

    }
}

在这里插入图片描述

lass Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        if(root==null){
            root=new TreeNode(val);
        }
        if(root.val>val){
            root.left=insertIntoBST(root.left,val);
        }
        if(root.val<val){
            root.right=insertIntoBST(root.right,val);
        }
        return root;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值