题目介绍
题目链接
236. 二叉树的最近公共祖先 - 力扣(LeetCode)
题目详情
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。
示例 1:
输入:root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1 输出:3 解释:节点5和节点1的最近公共祖先是节点3 。
示例 2:
输入:root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4 输出:5 解释:节点5和节点4的最近公共祖先是节点5 。 因为根据定义最近公共祖先节点可以为节点本身。
示例 3:
输入:root = [1,2], p = 1, q = 2 输出:1
思路讲解
一般来说想要找到两个节点的公共祖先节点,要从根节点往下走,这两个节点最后一次相交的地方就是这些节点的公共祖先。还有一种思路就是从两个节点往上找,一直走到根节点,则这两个节点到根节点的路程一定有相交的部分,即我们要的结果。
但是在写代码上并没有上面所说的那么容易,我们还得考虑好几种情况,具体如下:
- p 和 q 都在 root 的子树上,且分在两侧;
- p = root ,q 在 root 的(左或右)子树中;
- q = root ,p 在 root 的(左或右)子树中;
在具体的代码编写时又有不同的细节要考虑:
- 如果 left 和 right 都为null,说明 p 和 q 不在该子树 root 里,对于该子树直接返回 null;
- 如果Ieft和righti都不为nu,说明p和q分别在root的两侧;
- 当right为空,left不为空时,要考虑两种情况:(1)先判断一下root是不是p或者q,如果是说明q和p一个是另一个的祖先,直接返回就好了,否则:(2)说明ight子树里什么都没查到,而6和7是在lft子树里,此时需要递归的去左子树查即可。例如root为3时,此时递归的结果必然是right为null而left不为空。
- 如果left为空,而right不为空,说明是与3相反的情况。
具体实现
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if(root == null || 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 null;
if(left == null) return right;
if(right == null) return left;
return root;
}