算法 -【二叉树的最近公共祖先】

题目

给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。最近公共祖先的定义为：“对于有根树 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⁵] 内
• -10⁹ <= Node.val <= 10⁹
• 所有 Node.val 互不相同
• p != q
• p 和 q 均存在于给定的二叉树中

分析

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

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

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

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

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

代码

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;
}

原文：https://mp.weixin.qq.com/s/v-ZYi8MLXZj7QX5GvuNq3A