Leetcode 刷题第22天 | 235, 701

Leetcode 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, 因为根据定义最近公共祖先节点可以为节点本身。
说明:

  • 所有节点的值都是唯一的。
  • p、q 为不同节点且均存在于给定的二叉搜索树中。

算法思想:

  1. 递归法:
    递归参数:二叉搜索树根节点
    递归结束条件:如果当前root为None,返回None;如果当前root恰好为p(q)则返回p(q)[这里可能存在疑问,这里为什么就返回了?仔细读题,题目说明中说的很清楚,p,q均在二叉搜索树中。];最后一种情况就是,如果当前p(q)和q(p)分居root的左右两侧,则返回root,说明当前的root就是最近公共祖先。
    一次递归逻辑:递归遍历左子树,递归遍历右子树,别忘了层层返回。
  2. 迭代法:
    逻辑与递归法基本相似,二叉搜索树通常利用优秀的性质,采用中序遍历。但仔细想,本题是找到从下向上最近的公共节点,因此采用后序遍历。【为什么?后序遍历的顺序是左右中,因此可以模拟从下向上找的过程。】

代码实现:

  1. 递归法:
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def traversal(self, root, p, q):
        if root == p:
            return p
        if root == q:
            return q
        if root.val > p.val and root.val > q.val:
            left = self.traversal(root.left, p, q)
            if left:
                return left
        if root.val < p.val and root.val < q.val:
            right = self.traversal(root.right, p, q)
            if right:
                return right
        return root

    def lowestCommonAncestor(self, root, p, q):
        return self.traversal(root, p, q)
  1. 迭代法:
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def lowestCommonAncestor(self, root, p, q):
        while root:
            if root.val > p.val and root.val > q.val:
                root = root.left
            elif root.val < p.val and root.val < q.val:
                root = root.right
            else:
                return root
        return None

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

题目:

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

注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。
示例 1:
示例1
输入:root = [4,2,7,1,3], val = 5
输出:[4,2,7,1,3,5]
解释:另一个满足题目要求可以通过的树是:
示例1-1
示例 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]
提示:

  • 树中的节点数将在 [0, 1 0 4 10^4 104]的范围内。
  • - 1 0 8 10^8 108 <= Node.val <= 1 0 8 10^8 108
  • 所有值 Node.val 是 独一无二 的。
  • - 1 0 8 10^8 108 <= val <= 1 0 8 10^8 108
  • 保证 val 在原始BST中不存在。

算法思想:

  1. 递归法:
    按照二叉搜索树的条件,插入即可。
    要明确一点,二叉搜索树插入节点有多种方式
    但是向叶子节点插入数据是最简单的方式
    只要插入一个节点,就会在二叉搜索树的叶子节点位置找到插入的位置
  2. 迭代法:
    也是二叉搜索树的性质,也很简单

代码实现:

  1. 递归法:
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def traversal(self, root, p, q):
        if root == p:
            return p
        if root == q:
            return q
        if root.val > p.val and root.val > q.val:
            left = self.traversal(root.left, p, q)
            if left:
                return left
        if root.val < p.val and root.val < q.val:
            right = self.traversal(root.right, p, q)
            if right:
                return right
        return root

    def lowestCommonAncestor(self, root, p, q):
        return self.traversal(root, p, q)
  1. 迭代法:
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def lowestCommonAncestor(self, root, p, q):
        while root:
            if root.val > p.val and root.val > q.val:
                root = root.left
            elif root.val < p.val and root.val < q.val:
                root = root.right
            else:
                return root
        return None
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值