问题描述:
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 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。
示例 2:
输入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
输出: 5
解释: 节点 5 和节点 4 的最近公共祖先是节点 5。因为根据定义最近公共祖先节点可以为节点本身。
说明:
所有节点的值都是唯一的。
p、q 为不同节点且均存在于给定的二叉树中。
分析:
这道题需要在借助另一个方法才能做,这个方法就是在一个树中查找是否存在一个节点。
这个方法有四种情况:
1)这棵树如果是空树就直接返回false
2)判断根节点是否就是要找的这个节点
3)判断根节点的左子树中是否存在这个节点
4)判断根节点的左子树中是否存在这个节点
注意:判断根节点的左子树和右子树时可以运用递归来查找,如果找到了就返回true,反之返回false
接着就是找公共祖先的这个方法了,对这个方法,同样可以用递归
调用上面刚说的查找的方法,判断 p , q 是否存在于左子树中。如果 p 或者 q 就是根节点,那么就直接返回根节点 root ;如果全在左子树中,那么递归调用找公共祖先的这个方法继续查找距离最近的祖先(此时树的根节点会发生变化,变成 root . left );同样,如果全在右子树中,那么递归调用找公共祖先的这个方法继续查找距离最近的祖先(此时树的根节点会发生变化,变成 root . right ),剩余的情况就是 p 和 q 不在同一边 (无论 p 在哪一边都没有关系,肯定和 q 没在一边),那么就直接将root返回。
class Solution {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if (p == root || q == root){
return root;
}
boolean pleft = search(root.left,p);
boolean qleft = search(root.left,q);
//p,q都在左子树
if(pleft && qleft){
return lowestCommonAncestor(root.left,p,q);
}
//都在右子树
if( !pleft && !qleft){
return lowestCommonAncestor(root.right,p,q);
}
return root;
}
//查找q是否在p中
public static boolean search(TreeNode p, TreeNode q){
if (p == null){
return false;
}
if (p == q){
return true;
}
if(search(p.left,q)){
return true;
}
return search(p.right,q);
}
}