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
- 如果树中存在p、q则按题目定义;
- 如果树中只存在p,则p、q公共祖先是p;如果只存在q,p、q公共祖先是q;
- 如果树中p、q都不存在,则公共祖先是null(这条实际上可以被包括在2中);
以下所提到的“公共祖先”均代表“引申定义下的最近公共祖先”
递归函数终止条件:
找到公共祖先或者root为空
(实际上root==null也是一种找到公共祖先的情况定义3,可以去掉这个表述)
递归函数功能:
对于传入的树,找到p、q的公共祖先。(对每个根结点都是如此)
功能具体实现:
-
如果root等于p,返回p,因为:
-
如果树中有q,根据定义1,则p肯定是公共祖先,返回p
-
如果树中无q,根据定义2,则p也是公共祖先,仍然返回p
-
-
如果root等于q,同上返回q
-
如果root==null,说明p、q都不在树中(空树),根据定义3,返回null
-
如果以上不满足,说明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;
}
}
整理思路,记录博客,以便复习。若有误,望指正~