力扣刷题day19|235二叉搜索树的最近公共祖先、701二叉搜索树中的插入操作、450删除二叉搜索树中的节点

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

力扣题目链接

给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”

例如,给定如下二叉搜索树: root = [6,2,8,0,4,7,9,null,null,3,5]

示例 1:

输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 8
输出: 6 
解释: 节点 2 和节点 8 的最近公共祖先是 6。

示例 2:

输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 4
输出: 2
解释: 节点 2 和节点 4 的最近公共祖先是 2, 因为根据定义最近公共祖先节点可以为节点本身。

递归思路

那么本题是二叉搜索树,二叉搜索树是有序的‘

和二叉树:公共祖先问题 (opens new window)不同,普通二叉树求最近公共祖先需要使用回溯,从底向上来查找,二叉搜索树就不用了,因为搜索树有序(相当于自带方向),那么只要从上向下遍历就可以了。

其实只要从上到下遍历的时候,cur节点是数值在[p, q]区间中则说明该节点cur就是最近公共祖先了。

其他情况要么cur大于p和q或者cur小于p和q

递归三要素:

  1. 确定递归函数返回值以及参数:参数就是当前节点,以及两个结点 p、q。返回值为公共祖先TreeNode
TreeNode traversal(TreeNode root, TreeNode p, TreeNode q)
  1. 确定终止条件:找不到p、q就返回
if (root == null) {
    return null;
}
  1. 确定单层递归的逻辑

在遍历二叉搜索树的时候就是寻找区间[p.val, q.val](注意这里是左闭又闭)

那么如果 root.val 大于 p.val,同时 root.val 大于q.val,那么就应该向左遍历(说明目标区间在左子树上)。

if (root.val > p.val && root.val > q.val) {
    return traversal(root.left, p, q);
}

如果 root.val 小于 p.val,同时 root.val 小于 q.val,那么就应该向右遍历(目标区间在右子树)。

if (root.val < p.val && root.val < q.val) {
    return traversal(root.right, p, q);
}

剩下的情况,就是cur节点在区间(p.val <= root.val && root.val <= q.val)或者 (q.val <= root.val && root.val <= p.val)中,那么cur就是最近公共祖先了,直接返回cur。

return root;

完整代码:

public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
    return traversal(root, p, q);
}

TreeNode traversal(TreeNode root, TreeNode p, TreeNode q) {
    if (root == null) {
        return null;
    }
    if (root.val > p.val && root.val > q.val) {
        return traversal(root.left, p, q);
    }
    if (root.val < p.val && root.val < q.val) {
        return traversal(root.right, p, q);
    }
    return root;
}

701. 二叉搜索树中的插入操作

力扣题目链接

给定二叉搜索树(BST)的根节点 root 和要插入树中的值 value ,将值插入二叉搜索树。返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。

注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果

示例 1:

输入:root = [4,2,7,1,3], val = 5
输出:[4,2,7,1,3,5]

示例 2:

输入:root = [40,20,60,10,30,50,70], val = 25
输出:[40,20,60,10,30,50,70,null,null,25]

示例 3:

输入:root = [4,2,7,1,3,null,null,null,null,null,null], val = 5
输出:[4,2,7,1,3,5]

递归思路

这道题只考虑一种插入方式,就是按照二叉搜索树的规则去遍历,遇到空节点就插入目标节点,这样就不用调整二叉树的结构

递归三要素:

  1. 确定递归函数参数以及返回值:参数就是根节点指针,以及要插入元素。返回值是插入目标值后新的二叉树的根节点
public TreeNode insertIntoBST(TreeNode root, int val)
  1. 确定终止条件:就是找到叶子节点的下一个null插入,并把插入的节点返回。最下层返回的新树就是插入的这个新节点
// 叶子节点的下一个是null
if (root == null) {
    TreeNode node = new TreeNode(val);
    return node;
}
  1. 确定单层递归的逻辑:因为是搜索树,所以不用遍历整个数,根据大小关系确定向下遍历的方向
// 根据BST性质
if (val < root.val) {
    root.left = insertIntoBST(root.left, val);
}
if (val > root.val) {
    root.right = insertIntoBST(root.right, val);
}
return root;

完整代码:

public TreeNode insertIntoBST(TreeNode root, int val) {
    // 叶子节点的下一个是null
    if (root == null) {
        TreeNode node = new TreeNode(val);
        return node;
    }

    // 根据BST性质
    if (val < root.val) {
        root.left = insertIntoBST(root.left, val);
    }
    if (val > root.val) {
        root.right = insertIntoBST(root.right, val);
    }
    return root;
}

迭代思路

需要记录上一个节点(parent),遇到空节点了,就让parent左孩子或者右孩子指向新插入的节点

// 迭代--------------------------------------------------
public TreeNode insertIntoBST1(TreeNode root, int val) {
    // 叶子节点的下一个是null
    if (root == null) {
        TreeNode node = new TreeNode(val);
        return node;
    }
    TreeNode cur = root;
    TreeNode pre = null;
    // 找到叶子节点
    while (cur != null) {
        pre = cur;
        if (val < cur.val) {
            cur = cur.left;
        }else {
            cur = cur.right;
        }
    }

    TreeNode node = new TreeNode(val);
    if (val < pre.val) {
        pre.left = node;
    }else {
        pre.right = node;
    }

    return root;
}

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

力扣题目链接

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

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

  1. 首先找到需要删除的节点;
  2. 如果找到了,删除它。

示例 1:

输入:root = [5,3,6,2,4,null,7], key = 3
输出:[5,4,6,2,null,null,7]
解释:给定需要删除的节点值是 3,所以我们首先找到 3 这个节点,然后删除它。
一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。
另一个正确答案是 [5,2,6,null,4,null,7]。

示例 2:

输入: root = [5,3,6,2,4,null,7], key = 0
输出: [5,3,6,2,4,null,7]
解释: 二叉树不包含值为 0 的节点

示例 3:

输入: root = [], key = 0
输出: []

递归思路

递归三要素

  1. 确定递归函数参数以及返回值:参数就是根节点和要删除的值,返回值要是树节点,返回删除后的新树
public TreeNode deleteNode(TreeNode root, int key)
  1. 确定终止条件

要删除某个节点,就是遍历过程中找到删除节点,找到了就终止;没有找到就返回null终止

有以下五种情况:

  • 第一种情况:没找到删除的节点,遍历到空节点直接返回了

image-20221012192055919

// 第一种情况:没找到删除的节点,遍历到空节点直接返回了
if (root == null) return root;
  • 找到删除的节点

    • 第二种情况:左右孩子都为空(叶子节点),直接删除节点, 返回NULL为根节点

      image-20221012190439549

    // 第二种情况:左右孩子都为空(叶子节点),直接删除节点, 返回NULL为根节点
    if (root.left == null && root.right == null) {
        return null;
    }
    
    • 第三种情况:删除节点的左孩子为空,右孩子不为空,删除节点,右孩子补位,返回右孩子为根节点

    image-20221012191322053

    // 第三种情况:其左孩子为空,右孩子不为空,删除节点,右孩子补位 ,返回右孩子为根节点
    else if (root.left == null && root.right != null) {
        return root.right;
    }
    
    • 第四种情况:删除节点的右孩子为空,左孩子不为空,删除节点,左孩子补位,返回左孩子为根节点

    image-20221012191127099

    // 第四种情况:其右孩子为空,左孩子不为空,删除节点,左孩子补位,返回左孩子为根节点
    else if (root.left != null && root.right == null) {
        return root.left;
    }
    
    • 第五种情况:左右孩子节点都不为空,则将删除节点的左子树头结点(左孩子)放到删除节点的右子树的最左面节点的左孩子上,返回删除节点右孩子为新的根节点。

    image-20221012191649326

    // 第五种情况:左右孩子节点都不为空,则将删除节点的左子树放到删除节点的右子树的最左面节点的左孩子的位置
    // 并返回删除节点右孩子为新的根节点。
    else {
        // 先让cur指向要删除节点的右孩子
        TreeNode cur = root.right;
        while (cur.left != null) {
            cur = cur.left;
        }
        // 直到为左子树的叶子节点,把要删除的节点(root)左子树放在cur的左孩子的位置
        cur.left = root.left;
        // 返回旧root的右孩子作为新root
        return root.right;
    }
    
  1. 确定单层递归的逻辑:key小于根节点就往左找,大于根节点就往右找,根节点已经在前面的处理逻辑中处理过了

完整代码:

// 递归---------------------------------------------------
public TreeNode deleteNode(TreeNode root, int key) {
    // 第一种情况:没找到删除的节点,遍历到空节点直接返回了
    if (root == null) return root;

    if (root.val == key) {
        // 第二种情况:左右孩子都为空(叶子节点),直接删除节点, 返回NULL为根节点
        if (root.left == null && root.right == null) {
            return null;
        }
        // 第三种情况:其左孩子为空,右孩子不为空,删除节点,右孩子补位 ,返回右孩子为根节点
        else if (root.left == null && root.right != null) {
            return root.right;
        }
        // 第四种情况:其右孩子为空,左孩子不为空,删除节点,左孩子补位,返回左孩子为根节点
        else if (root.left != null && root.right == null) {
            return root.left;
        }
        // 第五种情况:左右孩子节点都不为空,则将删除节点的左子树放到删除节点的右子树的最左面节点的左孩子的位置
        // 并返回删除节点右孩子为新的根节点。
        else {
            // 先让cur指向要删除节点的右孩子
            TreeNode cur = root.right;
            while (cur.left != null) {
                cur = cur.left;
            }
            // 直到为左子树的叶子节点,把要删除的节点(root)左子树放在cur的左孩子的位置
            cur.left = root.left;
            // 返回旧root的右孩子作为新root
            return root.right;
        }
    }

    // 单层处理
    if (key < root.val) {
        root.left = deleteNode(root.left, key);
    }
    if (key > root.val) {
        root.right = deleteNode(root.right, key);
    }
    return root;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,关于力扣刷C++常用操作,我可以给你一些常见的操作和技巧: 1. 使用 STL 容器和算法库:STL(Standard Template Library)是 C++ 标准库中的一个重要组成部分,包含了许多常用的容器和算法。在力扣刷中,使用 STL 可以大大提高代码的效率和可读性。例如,vector 可以用来存储动态数组,sort 可以用来排序等等。 2. 使用 auto 关键字:auto 关键字可以自动推导变量类型,可以减少代码量和提高可读性。例如,auto x = 1; 可以自动推导出 x 的类型为 int。 3. 使用 lambda 表达式:lambda 表达式是 C++11 中引入的一种匿名函数,可以方便地定义一些简单的函数对象。在力扣刷中,使用 lambda 表达式可以简化代码,例如在 sort 函数中自定义比较函数。 4. 使用位运算:位运算是一种高效的运算方式,在力扣刷中经常会用到。例如,左移运算符 << 可以用来计算 2 的幂次方,右移运算符 >> 可以用来除以 2 等等。 5. 使用递归:递归是一种常见的算法思想,在力扣刷中也经常会用到。例如,二的遍历、链表的反转等等。 6. 使用 STL 中的 priority_queue:priority_queue 是 STL 中的一个容器,可以用来实现堆。在力扣刷中,使用 priority_queue 可以方便地实现一些需要维护最大值或最小值的算法。 7. 使用 STL 中的 unordered_map:unordered_map 是 STL 中的一个容器,可以用来实现哈希表。在力扣刷中,使用 unordered_map 可以方便地实现一些需要快速查找和插入的算法。 8. 使用 STL 中的 string:string 是 STL 中的一个容器,可以用来存储字符串。在力扣刷中,使用 string 可以方便地处理字符串相关的问。 9. 注意边界条件:在力扣刷中,边界条件往往是解决问的关键。需要仔细分析目,考虑各种边界情况,避免出现错误。 10. 注意时间复杂度:在力扣刷中,时间复杂度往往是评判代码优劣的重要指标。需要仔细分析算法的时间复杂度,并尽可能优化代码。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值