ACWing 88.树中两个节点的最低公共祖先（Python版）

题目描述

给出一个二叉树，输入两个树节点，求它们的最低公共祖先。
一个树节点的祖先节点包括它本身。

注意：
输入的二叉树不为空；
输入的两个节点一定不为空，且是二叉树中的节点；

样例：

解题思路

见注释

代码实现

class Solution(object):
    def lowestCommonAncestor(self, root, p, q):
        """
        :type root: TreeNode
        :type p: TreeNode
        :type q: TreeNode
        :rtype: TreeNode
        """
        if not root:
            return None
        if q == root or p == root: # 如果找到了某一个点 则当前节点为答案
            return root
        left = self.lowestCommonAncestor(root.left, p, q)
        right = self.lowestCommonAncestor(root.right, p, q)
        if left and right: # 如果左子树中可以找到其公共祖先 右子树也可以 则说明当前节点就是答案
            return root
        if not left: # 左边找不到了 那就是右子树的答案
            return right
        elif not right: # 右边亦然
            return left