二叉树的最近公共祖先

剑指 Offer 68 - II. 二叉树的最近公共祖先

题目描述

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

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

例如，给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4]

示例 1:

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

解法：递归

终止条件：

1. 当越过叶节点，则直接返回 null ；
2. 当 root 等于 p, q ，则直接返回 root ；

递推工作：

1. 开启递归左子节点，返回值记为 left ；
2. 开启递归右子节点，返回值记为 right ；

返回值： 根据 left 和 right ，可展开为四种情况；

1. 当 left 和 right 同时为空 ：说明 root 的左 / 右子树中都不包含 p,q ，返回null ；
2. 当 left 和 right 同时不为空 ：说明 p, q 分列在 root 的异侧 （分别在 左 / 右子树），因此 root 为最近公共祖先，返回 root ；
3. 当 left 为空，right 不为空 ：p,q 都不在 root 的左子树中，直接返回 right。具体可分为两种情况：
   • p,q 其中一个在 root 的 右子树 中，此时 right 指向 p（假设为 p ）；
   • p,q 两节点都在 root 的 右子树 中，此时的 right 指向 最近公共祖先节点 ；
4. 当left 不为空 ， rightt 为空 ：与情况 3. 同理；

var lowestCommonAncestor = function(root, p, q) {
  // 如果树为空，直接返回null
  if(root == null) return null; 
  // 如果 p和q中有等于 root的，那么它们的最近公共祖先即为root（一个节点也可以是它自己的祖先）
  if(root == p || root == q) return root; 
  // 递归遍历左子树，只要在左子树中找到了p或q，则先找到谁就返回谁
  let left = lowestCommonAncestor(root.left, p, q); 
  // 递归遍历右子树，只要在右子树中找到了p或q，则先找到谁就返回谁
  let right = lowestCommonAncestor(root.right, p, q); 
  // 如果在左子树中 p和 q都找不到，则 p和 q一定都在右子树中，右子树中先遍历到的那个就是最近公共祖先（一个节点也可以是它自己的祖先）
  if(left == null){
    return right;
  }else if(right == null){
// 否则，如果 left不为空，在左子树中有找到节点（p或q），这时候要再判断一下右子树中的情况，如果在右子树中，p和q都找不到，则 p和q一定都在左子树中，左子树中先遍历到的那个就是最近公共祖先（一个节点也可以是它自己的祖先）
  return left;
  }else{
  return root; //否则，当 left和 right均不为空时，说明 p、q节点分别在 root异侧, 最近公共祖先即为 root 
  } 
};