题目描述
给定一棵二叉树以及这棵树上的两个节点 o1 和 o2,请找到 o1 和 o2 的最近公共祖先节点。
解:
这道题可以采用递归查找+标记法求解,先找到两个节点的位置并用栈保存路径,(因为没有父节点指向没办法回滚)。然后对路径进行重合检查即可。
这道题有点特殊,返回值只能是int,也就是说不存在找不到的情况。所以边界检查可做可不做。
import java.util.*;
/*
* public class TreeNode {
* int val = 0;
* TreeNode left = null;
* TreeNode right = null;
* }
*/
public class Solution {
/**
*
* @param root TreeNode类
* @param o1 int整型
* @param o2 int整型
* @return int整型
*/
Stack<Integer> s1=new Stack<Integer>();
Stack<Integer> s2=new Stack<Integer>();
public boolean find(TreeNode ro,int o,Stack s){
if(ro==null)
return false;
s.push(ro.val);
if(ro.val==o){
return true;
}
if(find(ro.left,o,s)){
return true;
}
if(find(ro.right,o,s)){
return true;
}
s.pop();
return false;
}
public int lowestCommonAncestor (TreeNode root, int o1, int o2) {
int v1=0,v2=0;
if(!find(root,o1,s1)){
return 0;
}
//因为递归单次查找速度较快,这里就做两次查找了,保持代码简洁
if(!find(root,o2,s2)){
return 0;
}
//size()即树的层次,从1开始计算,最近祖节点层<=min(o1层,o2层)
while(s1.size()>s2.size()){
s1.pop();
}
while(s2.size()>s1.size()){
s2.pop();
}
while(!s1.empty() && !s2.empty()){
v1=s1.pop();
v2=s2.pop();
if(v1==v2){
return v1;
}
}
return 0;
}
}