LeetCode第236题

难度：中等

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

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

• 树中节点数目在范围 [2, 10^5] 内。

• -10^9 <= Node.val <= 10^9

• 所有 Node.val 互不相同 。

• p != q

• p 和 q 均存在于给定的二叉树中。

问题分析

这题让找出两个节点的最近公共节点，有两种解决方式，一种是从两个要查找的节点到根节点上的路径都连接起来，那么这两条路径就相当于两个链表了，这题就变成了查找两个链表的第一个公共节点了。

怎么连接呢？我们可以遍历这棵二叉树，然后使用一个map记录所遍历节点的父节点，然后在根据当前节点一直往上查找父节点，一直到根节点，这种方式比较简单，我们再来看另一种方式。

查找两个节点的最近公共节点，也就是从下往上找，我们知道二叉树都是从上往下遍历的，没法从下往上遍历。也就是说如果知道一个节点，肯定能找到他的子节点，但是我们没法找到他的父节点。

我们这里再来回顾一下递归，对于一棵二叉树来说因为是从根节点开始的，当递归往回走的时候不就是相当于从下往上遍历吗，所以这题我们可以参考二叉树的后序遍历来写。

1，如果两个节点都在左子树上，就返回左子树节点。

2，如果两个节点都在右子树上，就返回右子树节点。

3，如果两个节点分别在两棵子树上，说明当前节点就是他俩的最近公共祖先节点，直接返回当前节点即可。

JAVA：

public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
    if (root == null || p == root || q == root)
        return root;
    // 参考二叉树的后序遍历
    TreeNode left = lowestCommonAncestor(root.left, p, q);
    TreeNode right = lowestCommonAncestor(root.right, p, q);
    if (left == null)// 左子树为空，肯定都在右子树上
        return right;
    if (right == null)// 右子树为空，肯定都在左子树上
        return left;
    // 左右子树都不为空，一个在左子树一个在右子树，所以root就是他们的最近公共祖先节点。
    return root;
}

C++：

public:
    TreeNode *lowestCommonAncestor(TreeNode *root, TreeNode *p, TreeNode *q) {
        if (root == nullptr || p == root || q == root)
            return root;
        // 参考二叉树的后序遍历
        TreeNode *left = lowestCommonAncestor(root->left, p, q);
        TreeNode *right = lowestCommonAncestor(root->right, p, q);
        if (left == nullptr)// 左子树为空，肯定都在右子树上
            return right;
        if (right == nullptr)// 右子树为空，肯定都在左子树上
            return left;
        // 左右子树都不为空，一个在左子树一个在右子树，所以root就是他们的最近公共祖先节点。
        return root;
    }

C：

struct TreeNode *lowestCommonAncestor(struct TreeNode *root, struct TreeNode *p, struct TreeNode *q) {
    if (root == NULL || p == root || q == root)
        return root;
    // 参考二叉树的后序遍历
    struct TreeNode *left = lowestCommonAncestor(root->left, p, q);
    struct TreeNode *right = lowestCommonAncestor(root->right, p, q);
    if (left == NULL)// 左子树为空，肯定都在右子树上
        return right;
    if (right == NULL)// 右子树为空，肯定都在左子树上
        return left;
    // 左右子树都不为空，一个在左子树一个在右子树，所以root就是他们的最近公共祖先节点。
    return root;
}

Python：

def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
    if root is None or p == root or q == root:
        return root
    # 参考二叉树的后序遍历
    left = self.lowestCommonAncestor(root.left, p, q)
    right = self.lowestCommonAncestor(root.right, p, q)
    if left is None:  # 左子树为空，肯定都在右子树上
        return right 
    if right is None: # 右子树为空，肯定都在左子树上
        return left
    # 左右子树都不为空，一个在左子树一个在右子树，所以root就是他们的最近公共祖先节点。
    return root