ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

《剑指 Offer》学习记录:题 68:二叉树的最近公共祖先

2021-07-24 01:32:11  阅读:194  来源: 互联网

标签:结点 val Offer 祖先 二叉树 公共 68 root 节点


目录

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

题干

给定一个二叉搜索树,找到该树中两个指定节点的最近公共祖先。最近公共祖先是对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。
例如对于如图的二叉搜索树,节点 1 和节点 5 的最近公共祖先是节点 3。节点 5 和节点 4 的最近公共祖先是节点 5,因为根据定义最近公共祖先节点可以为节点本身。

二叉树结点的定义是:

class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

解题思路

由于是二叉搜索树,所以可以利用二叉搜索树的性质来解题。对于二叉搜索树来说,满足条件的结点的数据域,值是 2 个给定数据的中间值。例如对于结点 2 和 5,他们的公共祖先是结点 3,从数值上看 3 是 2 和 5 的中间值。不过为什么不能是结点 4 呢?因为结点 4 虽然是中间值,但是却不是第一个遇到的中间值,也就是结点 3 所在的层次比其更上层。进行搜索操作时,结点 3 就会被先遍历到。

因此若结点比 2 个给定数据都大,则最近公共祖先会出现在结点的左子树中,若结点比 2 个给定数据都小,则最近公共祖先会出现在结点的右子树中。同理对于结点 8 和 7,其公共祖先就是结点 8,因为结点 8 本身就可以访问它本身。

根据对题目的分析,可以得到解题的伪代码如下。

题解代码

解法的实现可以递归也可以迭代,思路是一样的。

递归

class Solution:
    def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
        if root.val > p.val and root.val > q.val:
            return self.lowestCommonAncestor(root.left, p, q)
        elif root.val < p.val and root.val < q.val:
            return self.lowestCommonAncestor(root.right, p, q)
        else:
            return root

迭代

class Solution:
    def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
        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:
                break
        return root

时空复杂度

由于从本质上来看,这题其实也是二叉搜索树的查找操作,时间复杂度 O(㏒2n)。
如果使用迭代法,只需要常数级别的存储空间,空间复杂度为 O(1)。如果使用递归法法,最差情况下要把所有结点都递归一遍,空间复杂度为 O(n)。

普通二叉搜的最近公共祖先

给定一个普通二叉树,找到该树中两个指定节点的最近公共祖先。
例如对于如图的二叉树,节点 1 和节点 3 的最近公共祖先是节点 7。节点 8 和节点 4 的最近公共祖先是节点 4,因为根据定义最近公共祖先节点可以为节点本身。

解题思路

因为此时不是二叉搜索树,所以不能用二叉搜索树的性质解决。解决这道题需要使用 DFS,对于每个结点在这 3 种情况下直接返回结点本身。

  1. 当前结点是 p;
  2. 当前结点是 q;
  3. 当前结点是 null;

若不属于以上 3 种情况则进行 DFS,DFS 的结果有 4 种:

  1. p、q 都不在当前结点的子树中,说明没有找到最近公共祖先,返回 null;
  2. p 在当前结点的子树中,返回 p
  3. q 在当前结点的子树中,返回 q
  4. p、q 都在当前结点的子树中,说明当前结点就是最近公共祖先,返回结点本身;

题解代码

class Solution:
    def lowestCommonAncestor(self, root: TreeNode, p: TreeNode, q: TreeNode) -> TreeNode:
        #当前结点为空
        if root == None: 
            return None
        #当前结点是要找的结点之一
        if root == p or root == q: 
            return root
        left = self.lowestCommonAncestor(root.left, p, q)
        right = self.lowestCommonAncestor(root.right, p, q)
        #左右子树都没找到
        if left == None and right == None: 
            return
        #在右子树找到了
        if left == None: 
            return right
        #在左子树找到了
        if right == None: 
            return left
        #当前结点是最近公共祖先
        return root

时空复杂度

最差情况下要把所有结点都搜索一遍,空间复杂度和空间复杂度都为 O(n)。

参考资料

《剑指 Offer(第2版)》,何海涛 著,电子工业出版社
面试题68 - I. 二叉搜索树的最近公共祖先(迭代 / 递归,清晰图解)
剑指68II:最近公共祖先2解:后序遍历,路径回溯

标签:结点,val,Offer,祖先,二叉树,公共,68,root,节点
来源: https://www.cnblogs.com/linfangnan/p/15046625.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有