(二刷)代码随想录第22天|450.删除二叉搜索树中的节点 235. 二叉搜索树的最近公共祖先

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

递归三部曲:

1、确定参数和返回值:

class Solution{
    public TreeNode deleteNode(TreeNode root, int key){
        
    }

}

2、确定终止条件

if(root == null){
    return root;
}

3、确定单层遍历的逻辑:

//如果当前值等于要删除的值
if(root.val == key){
    //如果当前节点没有左子节点,直接返回右子节点
    if(root.left == null){
        return root.right;
    }
    //如果当前节点没有右子节点,直接返回左子节点
    else if(root.right == null){
        return root.left;
    }
    //如果当前节点有2个子节点
    else{
        //找到右子树最左边的节点
        TreeNode cur = root.right;
        while(cur.left != null){
            cur=cur.left;
        }
        //将要删除节点的左子树连接到右子树最左边的节点上
        cur.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;

235. 二叉搜索树的最近公共祖先

class Solution {
    // 定义一个方法,用于在二叉搜索树中寻找两个节点的最近公共祖先
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        // 使用while循环进行遍历
        while (true) {
            // 如果当前节点的值大于p和q的值,则最近公共祖先在左子树中
            if (root.val > p.val && root.val > q.val) {
                root = root.left; // 移动到左子节点
            // 如果当前节点的值小于p和q的值,则最近公共祖先在右子树中
            } else if (root.val < p.val && root.val < q.val) {
                root = root.right; // 移动到右子节点
            // 如果当前节点的值介于p和q的值之间,或者等于p或q的值,说明当前节点就是最近公共祖先
            } else {
                break; // 退出循环
            }
        }
        // 返回最近公共祖先节点
        return root;
    }
}
class Solution {
    // 定义一个方法,用于在二叉搜索树中插入一个新的节点
    public TreeNode insertIntoBST(TreeNode root, int val) {
        // 如果根节点为空,则创建一个新的节点作为根节点并返回
        if (root == null) return new TreeNode(val);
        
        // 保存根节点的引用,以便最后返回
        TreeNode newRoot = root;
        // 定义一个指针用于跟踪前一个节点
        TreeNode pre = root;
        
        // 使用while循环寻找插入位置
        while (root != null) {
            // 将当前节点赋值给pre
            pre = root;
            // 如果当前节点的值大于插入值,则移动到左子节点
            if (root.val > val) {
                root = root.left;
            // 如果当前节点的值小于插入值,则移动到右子节点
            } else if (root.val < val) {
                root = root.right;
            } 
        }
        
        // 找到插入位置后,根据pre节点的值决定插入在左子节点还是右子节点
        if (pre.val > val) {
            pre.left = new TreeNode(val);
        } else {
            pre.right = new TreeNode(val);
        }

        // 返回更新后的根节点
        return newRoot;
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值