leetcode236.二叉树的最近公共祖先 Python

题目：

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

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

示例：

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

思路：

• 找最近的公共祖先，用递归的思路是首先传入根节点，如果根节点为空，或者根节点就是p节点或者q节点，就直接返回根节点即可，这就是最近公共祖先了。
• 然后递归，在根节点的左子树上找最近公共祖先，返回值，就是在左子树上的查找情况，在根节点的右子树上找最近公共祖先，返回值，就是在右子树上的查找情况，通过递归调用，遍历整个二叉树。
• 最后如果没有在一开始就返回，就是根节点不是p或者q节点，这时候如果左子树返回值是空，那祖先就在右子树的返回值上，如果右子树返回值是空，那祖先就在左子树的返回值上，如果两个返回值都不为空，在左右子树上都没有找到最近公共祖先，那最近公共祖先就是根节点，返回根节点即可。

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution(object):
    def lowestCommonAncestor(self, root, p, q):
        """
        :type root: TreeNode
        :type p: TreeNode
        :type q: TreeNode
        :rtype: 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:
            return right
        if not right:
            return left
        return root