【剑指offer】树中两个结点的最低公共祖先

题目链接

acwing
leetcode

题目描述

给出一个二叉树，输入两个树节点，求它们的最低公共祖先。
一个树节点的祖先节点包括它本身。

题目思路

1. 需要回溯，将公共祖先结果返回，所以采用后序遍历：左右中，只有后序遍历才向上回溯
2. 后序遍历的处理过程是从下往上；如果想从下往上处理，就需要后序遍历
3. 判断左子树是否出现了p和q，反应在递归上：为终止条件，如果出现了，就向上返回。
4. 中的处理逻辑为：如果左子树的返回值不为空，可能出现了p或q，右子树的返回值不为空，出现了q或p。左不为空，右不为空，则当前的中一定是公共祖先（左子树有p或q，右子树有q或p）

---

1. 递归返回值，传入参数

   1. 返回值：公共祖先
   2. 传入参数：根节点，p，q

2. 终止条件

   1. 如果root为空，返回null；如果遍历空的话，就终止了
   2. 如果遇到p或q了，需要向上返回，告诉上层节点遇到p或q了，返回当前节点

3. 单层递归逻辑

   1. 左
      1. 向左遍历
      2. 返回值告诉我们左子树中是否出现了p或q
   2. 右
      1. 向右遍历
      2. 返回值告诉我们右子树是否出现了p或q
   3. 中
      1. 如果left和right都不为空，说明左子树出现了p或q，右子树出现了q或p。说明当前的root就是最近公共祖先，将root返回即可
      2. 如果left为空，right不为空，应该将right继续向上返回。顶部才能得到节点right（此时right为公共祖先）
      3. 如果left不为空，right为空，应该将left向上返回，这样顶部才能得到left
      4. 如果left和right都为空，返回null

4. 如果p是公共祖先 或 如果q是公共祖先

   1. 这种情况包含在了上述的代码中
   2. 因为遇到了p或q就返回了，就不继续向下遍历了

代码实现

  /**
     * Definition for a binary tree node.
     * public class TreeNode {
     *     int val;
     *     TreeNode left;
     *     TreeNode right;
     *     TreeNode(int x) { val = x; }
     * }
     */
    class Solution {
        //后序
        public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
            //终止条件
            if(root == null) return null;
            //如果遇到了p或q，向上返回
            if(root == p || root == q) return root;
    
            //左
            TreeNode left = lowestCommonAncestor(root.left, p, q);
    
            //右
            TreeNode right = lowestCommonAncestor(root.right, p, q);
    
            //中
            //当前节点为公共祖先
            if(left != null && right != null){
                return root;
            }
            //这样left为公共祖先，返回left，顶部才能得到left
            else if(left != null && right == null){
                return left;
            }
            else if(left == null && right != null){
                return right;
            }
            //两个都为空，没有公共祖先
            else{
                return null;
            }
        }
    }