leetcode 236. 二叉树的最近公共祖先

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

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

例如，给定如下二叉树:  root = [3,5,1,6,2,0,8,null,null,7,4]

---

示例 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。因为根据定义最近公共祖先节点可以为节点本身。

---

暴力递归

# 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 self.dfs(root,p) or not self.dfs(root,q):
            return None
        if root == p or root == q:
            return root
        else:
            pleft = self.dfs(root.left,p)
            qleft = self.dfs(root.left,q)
            if pleft == qleft == True:
                return self.lowestCommonAncestor(root.left,p,q)
            elif pleft == qleft == False:
                return self.lowestCommonAncestor(root.right,p,q)
            else:
                return root

    
    def dfs(self, root, node):
        if not root:
            return False
        if root == node:
            return True
        left = self.dfs(root.left,node)
        right = self.dfs(root.right,node)
        return left or right

最直观的递归，分类讨论p和q分别是在左子树还是右子树，还是root等于其中的一个。如果其中一个与当前的node相等，那么这个node必为最近公共祖先。如果pq分别在左右子树，这个node也为最近公共祖先。但是如果pq都在其左子树上或者右子树上，那么需要进一步的递归。

改进版dfs

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

根据定义，如果节点root是p和q的最近公共祖先，那么一定满足以下其中一种情况：1.p和q在root的子树中，且分居两侧。2.root就是p节点且q在其子树中。3.root是q节点，且p在其子树中。所以我们可以使用后序遍历，分别获得root左右子树的情况，然后如果root为q或p，或者说p和q分居其两侧，那么就返回这个root。

考虑边界条件：如果为叶子节点，我们需判断是否为p或者q，如果是那么就返回root节点，不然就要递归它的left和right，但显然作为叶子节点其左右儿子都为空，那么当为空的时候就返回空。所以对于一个叶子节点来讲有两种返回情况，一种为root另一种为None。

考虑返回情况：

根据left和right一共有四种情况：1.left和right都不为空，那么说明pq在两侧，返回root。2.left和right都为空，那么说明root的左右子树不含其节点，返回None（这里不需要考虑root是否和pq相等因为已经在入口处进行判断，如果相等直接返回root）3.left为空，right不为空，那么直接返回right，但是这里right有两种可能，一种是最近公共节点在right， 另一种是right子树包含了一个p或者q。4和3相反。

注：本题还可以使用哈希表来解，哈希表中保存每个节点的父节点。