给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分三种情况:
- 如果这两个节点其中一个是根节点,直接返回根节点
- 如果一个结点在左子树,另一个在右子树,直接返回根节点
- 如果两个都在左子树或者都在右子树上,直接使用递归返回最近公共祖先。
class Solution {
public boolean search(TreeNode r,TreeNode t){
if(r==null){
return false;
}
if(r==t){
return true;
}
if(search(r.left,t)){
return true;
}
if(search(r.right,t)){
return true;
}
return false;
}
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if(p==root||q==root){
return root;
} //这两个节点至少一个是根节点
boolean ph=search(root.left,p);
boolean pf=search(root.right,q);
boolean phi=search(root.right,p);
boolean pfi=search(root.left,q);
if(ph&&pf||phi&&pfi){
return root;
} // 一个在左子树,一个在右子树
if(ph&&pfi){
return lowestCommonAncestor(root.left,p,q); //两个都在左子树
}
return lowestCommonAncestor(root.right,p,q); //两个都在右子树
}
}
这个算法时间复杂度较高,相对节省空间。