二叉树的最近公共祖先 （Python）

LeetCode链接

后续遍历 递归

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
        '''
        node为p、q最近公共祖先的三种情况:
        1、p、q在node的左右子树中
        2、p == node 且 q在node的左或右子树中
        3、q == node 且 p在node的左或右子树中
        则 先要知道节点左右子树的情况（是否包含p、q）之后才能决定如何上升
        使用后序遍历 递归
        '''
        def doit(root):
            # root为None 则子树上没有pq 返回None 若子树上有pq返回pq的值
            if not root or root == p or root == q: return root
            # 递归查看左右子树是否包含pq
            left = doit(root.left)
            right = doit(root.right)
            # 所有Node.val 均不同 且p!=q
            # 若左右子树均有值返回 则说明 pq分别存在于当前节点的左右子树中 则最近公共祖先就是当前节点
            if left and right: 
                return root
            else:
                # 左边不存在pq 则看右边 反之相同 若都不存在 返回None
                return left or right
        return doit(root)