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

目录

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

题干

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

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 就会被先遍历到。
《剑指 Offer》学习记录:题 68:二叉树的最近公共祖先
因此若结点比 2 个给定数据都大,则最近公共祖先会出现在结点的左子树中,若结点比 2 个给定数据都小,则最近公共祖先会出现在结点的右子树中。同理对于结点 8 和 7,其公共祖先就是结点 8,因为结点 8 本身就可以访问它本身。
《剑指 Offer》学习记录:题 68:二叉树的最近公共祖先
根据对题目的分析,可以得到解题的伪代码如下。
《剑指 Offer》学习记录:题 68:二叉树的最近公共祖先

题解代码

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

递归

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)。
《剑指 Offer》学习记录:题 68:二叉树的最近公共祖先

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

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

解题思路

因为此时不是二叉搜索树,所以不能用二叉搜索树的性质解决。解决这道题需要使用 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》学习记录:题 68:二叉树的最近公共祖先

参考资料

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

上一篇:力扣 68. 文本左右对齐


下一篇:FirefoxOS体验与Web开发