LeetCode 236 [Lowest Common Ancestor of a Binary Tree]

原题

给定一棵二叉树，找到两个节点的最近公共父节点(LCA)。
最近公共祖先是两个节点的公共的祖先节点且具有最大深度。

对于下面这棵二叉树

  4
 / \
3   7
   / \
  5   6
LCA(3, 5) = 4
LCA(5, 6) = 7
LCA(6, 7) = 7

解题思路

• Divide & Conquer 的思路
• 如果root为空，则返回空
• 如果root等于其中某个node，则返回root
• 如果上述两种情况都不满足，则divide，左右子树分别调用该方法
• Divide & Conquer中治这一步要考虑清楚，本题三种情况
• 如果left和right都有结果返回，说明root是最小公共祖先
• 如果只有left有返回值，说明left的返回值是最小公共祖先
• 如果只有�right有返回值，说明�right的返回值是最小公共祖先

完整代码

# 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:
            return None
        if root == p or root == q:
            return root
        
        # divide
        left = self.lowestCommonAncestor(root.left, p, q)
        right = self.lowestCommonAncestor(root.right, p, q)
        
        # conquer
        if left != None and right != None:
            return root
        if left != None:
            return left
        else:
            return right