12 LeetCode 236题 二叉树的最近公共祖先

LeetCode 236题 二叉树的最近公共祖先

1. 题目描述

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

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

此题的条件至关重要（尤其是**Node.val**互不相同，不然递归遍历无法求解）

2. 题目分析

题目已经给了最近公共祖先的定义，但是可能大家还是比较“懵”，我们通过上面的例图来进行举例说明

先来举两个普通的例子：

• 节点5 1的最近公共祖先为 3（5的祖先为 5 3， 1的祖先为 1 3，所以其最近公共祖先为**3**）
• 节点6 2的最近公共祖先为 5 （6的祖先为 6 5 3， 2的祖先为 2 5 3，所以其最近公共祖先为**5**）

接着我们看看特殊的例子：

• 节点5 6的最近公共祖先为 5（因为一个节点也可以是它自己的祖先，5的祖先为 5 3， 6的祖先为 6 5 3，所以其最近公共祖先为5）
• 节点2 0的最近公共祖先为 3 （2的祖先为2 5 3， 0的祖先为0 1 3，所以其最近公共祖先为**3**）

3. 总结

通过上述举例，大家应该已经了解了什么是 最近公共祖先，那么我们该如何来用代码判断呢？

我们可以通过后续遍历的方式递归，如上图

• 我们想找到节点6 2的最近公共祖先5，我们采用后续遍历发现6和2分别在以5为根节点的左右子树中
• 我们想找到节点5 6的最近公共祖先5，我们采用后续遍历发现5和6分别在以5为根节点的根节点与左子树中
• 我们想找到节点5 2的最近公共祖先5，我们采用后续遍历发现5和2分别在以5为根节点的根节点与右子树中

综上，我们发现，如果想找到两个节点的最近公共祖先，只需要满足一下三种条件的任意两种

• p q 分别存在于 根节点 左子树
• p q 分别存在于 根节点 右子树
• p q 分别存在于 左子树 右子树

我们在递归中通过以上三种方式判断即可

4. 代码

代码如下：

//初始化公共祖先为空
private TreeNode lca = null;

public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q)
{
    //如果树为空，直接返回空
    if (root == null)
    {
        return null;
    }
    findNode(root, p, q);
    return lca;
}

/**
 * @param root
 * @param p
 * @param q
 * @return 看从root出发能不能找到p或者q，只要找到一个，就返回true，都找不到的话返回false；
 */
private boolean findNode(TreeNode root, TreeNode p, TreeNode q)
{
    if (root == null)
    {
        return false;
    }
    //按照后续遍历查找(找到返回1)（题目已经说明，各个节点值不相同且一定存在要找到的p、q，所以在左右子树中一定可以递归找到）
    //为方便下面判断当前节点是否为最近公共节点，将其设置为0 / 1
    int left = findNode(root.left, p, q) ? 1 : 0;
    int right = findNode(root.right, p, q) ? 1 : 0;
    int mid = (root == p || root == q) ? 1 : 0;
    //如果在其中的两个位置都找到了（满足以上条件中任意的两种），那说明root就是其最近公共节点
    if (left + right + mid == 2)
    {
        lca = root;
    }
    //如果其中有一个找到，返回true
    return left + right + mid > 0;
}