题目描述
给定一棵二叉树以及这棵树上的两个节点 o1 和 o2,请找到 o1 和 o2 的最近公共祖先节点。
解题思路
考虑用递归求解:
首先考虑终止条件,假设如下只有三个节点的树
存在两种情况:
1)根节点为给定的两个节点中的一个,则直接返回该节点
2)给定的两个节点分别位于根节点左右两侧,也返回根节点
进一步推广到多个节点的树如下:
那么递归遍历每个节点,将每个节点作为根节点,同以上有两种情况:
1)要么返回该根节点(给定两个节点位于该节点的左右子节点中);
2)要么返回该节点的子节点中的某一个子节点(该子节点正好为给定的两个节点中的一个)。
如果所有的子节点都向上返回它的父节点,也就是父节点继续向上返回它的父节点,最终会返回到整棵树的根节点。
程序如下:
public int lowestCommonAncestor (TreeNode root, int o1, int o2) {
// write code here
if(root == null){
return 0;
}
// 对应一个子节点正好是给定的两个节点中的一个
if(root.val == o1 || root.val == o2){
return root.val;
}
// 递归遍历所有的节点
int left = lowestCommonAncestor(root.left,o1,o2);
int right= lowestCommonAncestor(root.right,o1,o2);
// 如果某一侧的子节点为空则执行它的右侧节点,保证遍历每一个节点。
if(left == 0){
return right;
}
if(right == 0){
return left;
}
// 所有子节点都向上返回它的父节点,最终到达根节点,也就是给定的两个节点位于根节点的两侧
return root.val;
}