leetcode----236.二叉树的最近公共祖先（后序遍历）

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

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

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

示例：

输入：root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
输出：3
解释：节点 5 和节点 1 的最近公共祖先是节点 3 。

思路：

• 后序遍历 参考这篇题解，有助于理解下面递归函数的定义

最近公共祖先的引申定义：
对于两个节点p、q和一棵树root

1. 如果树中存在p、q则按题目定义；
2. 如果树中只存在p，则p、q公共祖先是p；如果只存在q，p、q公共祖先是q；
3. 如果树中p、q都不存在，则公共祖先是null（这条实际上可以被包括在2中）；

以下所提到的“公共祖先”均代表“引申定义下的最近公共祖先”

递归函数终止条件：
找到公共祖先或者root为空
（实际上root==null也是一种找到公共祖先的情况定义3，可以去掉这个表述）

递归函数功能：

对于传入的树，找到p、q的公共祖先。（对每个根结点都是如此）

功能具体实现：

1. 如果root等于p，返回p，因为：

   • 如果树中有q，根据定义1，则p肯定是公共祖先，返回p

   • 如果树中无q，根据定义2，则p也是公共祖先，仍然返回p

2. 如果root等于q，同上返回q

3. 如果root==null，说明p、q都不在树中（空树），根据定义3，返回null

4. 如果以上不满足，说明p、q要么在子树中，要么不在树中，对root左右子节点调用函数

   • 如果右子树为空，说明公共祖先在左子树中，会通过返回值呈递上来
   • 如果左子树为空，说明公共祖先在右子树中，会通过返回值呈递上来
   • 如果都为空，实际上也包含在上面两种情况中，返回null
   • 如果都不为空，说明p、q分别在左右子树中，那么根节点root就是p、q公共祖先

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if(root == null || root.val == p.val || root.val == q.val){
            return root;
        }

        TreeNode left = lowestCommonAncestor(root.left, p, q);
        TreeNode right = lowestCommonAncestor(root.right, p, q);

        //l、r 非空时，说明 p、q 分居 root 的两侧，root 就是 LCA
        //l、r 任一为空，说明 LCA 位于另一子树或其祖先中
        
        if(left != null && right != null){
            return root;
        }
        return left == null ? right : left;
    }
}

整理思路，记录博客，以便复习。若有误，望指正~