题目
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
方法一:两次遍历
p 和 q 的最近公共祖先就是从根节点到它们路径上的「分岔点」,也就是最后一个相同的节点。设从根节点到 p 的路径为数组 path_p,从根节点到 q 的路径为数组 path_q,那么只要找出最大的编号 i,其满足path_p[i]=path_q[i]即可找出对应节点。
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
List<TreeNode> pathp=getPath(root,p);
List<TreeNode> pathq=getPath(root,q);
TreeNode an=null;
for(int i=0;i<pathp.size()&&i<pathq.size();++i){
if(pathp.get(i)==pathq.get(i)){
an=pathp.get(i);
}
else{
break;
}
}
return an;
}
public List<TreeNode> getPath(TreeNode root,TreeNode target){
List<TreeNode> path =new ArrayList<TreeNode>();
TreeNode node=root;
while(node!=target){
path.add(node);
if(target.val<node.val){
node=node.left;
}else{
node=node.right;
}
}
path.add(node);
return path;
}
复杂度分析
时间复杂度:O(n),其中 nn 是给定的二叉搜索树中的节点个数。
空间复杂度:O(n),我们需要存储根节点到 p 和 q 的路径。
方法二:一次遍历
遍历过程与方法一中的类似,根据当前节点值判断。
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
TreeNode res=root;
while(true){
if(p.val<res.val&&q.val<res.val){
res=res.left;
}else if(p.val>res.val&&q.val>res.val){
res=res.right;
}else{
break;
}
}
return res;
}
复杂度分析
时间复杂度:O(n),其中 n 是给定的二叉搜索树中的节点个数。
空间复杂度:O(1)。