LeetCode 热题 HOT 100 第五十九天 236. 二叉树的最近公共祖先 中等题 用python3求解

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

示例 1：

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

示例 2：

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

示例 3：
输入：root = [1,2], p = 1, q = 2
输出：1

提示：
树中节点数目在范围 [2, 10^5] 内。
-10的9次方 <= Node.val <= 10的9次方
所有 Node.val 互不相同 。
p != q
p 和 q 均存在于给定的二叉树中。


解法：DFS
参考：指路

祖先的定义： 若节点p在节点root 的左（右）子树中，或p=root ，则称root是p的祖先。

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

根据以上定义，若root是p,q的最近公共祖先 ，则只可能为以下情况之一：
1.p和q在root的子树中，且分列root 的异侧（即分别在左、右子树中）；
2.p=root，且q在root的左或右子树中；
3.q = root，且p在root的左或右子树中；

考虑通过递归对二叉树进行先序遍历，当遇到节点p或q时返回。从底至顶回溯，当节点p,q在节点root的异侧时，节点root即为最近公共祖先，则向上返回root 。

递归解析：

1.终止条件：
1.1当越过叶节点，则直接返回null；
1.2当root等于p,q，则直接返回root；

2.递推工作：
2.1开启递归左子节点，返回值记为left；
2.2开启递归右子节点，返回值记为right；

**3.返回值：**根据left和right，可展开为四种情况；

3.1当left和right同时为空：说明root的左 / 右子树中都不包含p,q，返回null；

3.2当left和right同时不为空：说明p,q分列在root 的异侧（分别在左/右子树），因此root为最近公共祖先，返回root；

3.3当left为空，right不为空：p,q都不在root的左子树中，直接返回right。具体可分为两种情况：
（1）p,q其中一个在root的 右子树 中，此时right指向p（假设为p）；
（2）p,q两节点都在root的 右子树 中，此时的right指向最近公共祖先节点 ；

3.4当left不为空，right为空：与情况3.同理；

观察发现， 情况 1. 可合并至 3. 和 4. 内，详见文章末尾代码。

复杂度分析：
时间复杂度O(N) ：其中N为二叉树节点数；最差情况下，需要递归遍历树的所有节点。
空间复杂度O(N) ：最差情况下，递归深度达到N，系统使用O(N)大小的额外空间。

代码实现：

class Solution:
    def lowestCommonAncestor(self, root: TreeNode, p: TreeNode, q: TreeNode) -> TreeNode:
        if not root 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 and not right: return # 第一种情况 可省略
        if not left: return right # 第三种情况
        if not right: return left # 第四种情况
        return root # 第二种情况 if left and right