Leetcode 236. 二叉树的最近公共祖先

题目

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

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

递归解法

最近公共祖先的定义： 设节点root为节点 p，q的某公共祖先，若其左子节点 root.left和右子节点 root.right都不是 p,q的公共祖先，则称root是“最近的公共祖先” 。

根据以上定义：若root满足最近的公共祖先，则

1. p、q在root的子树内，且分为两侧；
2. root=p，且q在root的左子树或右子树内；
3. root=q，且p在root的左子树或右子树内；

思路

可以通过递归对二叉树进行后续遍历，当遇到节点p或q时则返回，从低至上回溯，当节点p、q在节点root的异侧时，节点root则为最近公共祖先，返回root。

递归解析

1. 终止条件：
   1. 当越过叶节点，则直接返回 null ；
   2. 当 root等于 p, q，则直接返回 root；
2. 递推工作：
   1. 开启递归左子节点，返回值记为 left；
   2. 开启递归右子节点，返回值记为 right；
3. 返回值： 根据 left和 right，可展开为四种情况；
   1. 当 left和 right同时为空 ：说明 root的左 / 右子树中都不包含p, q，返回 null ；
   2. 当 left和 right同时不为空 ：说明p, q分列在root的 异侧 （分别在 左 / 右子树），因此 root为最近公共祖先，返回 root；
   3. 当 left为空 ，right不为空 ：p, q都不在 root的左子树中，直接返回right。具体可分为两种情况：
      p,q 其中一个在 root的 右子树 中，此时 right指向 p（假设为 p）；
      p,q 两节点都在 root的 右子树 中，此时的right指向最近公共祖先节点 ；
   4. 当 left不为空 ， right为空 ：与情况 3. 同理；

题解

class Solution:
    def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
        if root == None or root == p or root == q:
            return root
        left = self.lowestCommonAncestor(root.left, p, q)
        right = self.lowestCommonAncestor(root.right, p, q)
        if not left:
            return right
        if not right:
            return left
        return root