题目
给定一颗二叉树的头结点,和这颗二叉树中2个节点n1和n2,求这两个节点的最近公共祖先;
思路
利用后序遍历实现;
对于当前节点cur,如果节点为null或者等于n1或n2中的一个,则直接返回cur;
先处理左右子树,左子树返回left,右子树返回right;判断left和right;
1)left和right均为null,说明以cur为根的树上没发现n1和n2;
2)left和right均不为null,说明在左子树上发现了n1或n2,在右子树上也发现了n1或n2,cur为n1和n2的首次相遇节点,则直接返回cur;
3)left和right中一个null,另一个不为null,说明不为空的节点是n1和n2的其中一个,或者是n1和n2的最近公共祖先;则返回不为空的节点;
代码
public class LowestAncestor {
/*求解n1和n1的最近公共祖先
*
* 利用后序遍历求解
*/
public Node getLowestAncestor(Node head,Node n1,Node n2){
if(head==null || head==n1 || head==n2){
return head;
}
//先遍历左右子树,左右子树的返回为left和right;然后判断left和right请情况
Node left=getLowestAncestor(head.left,n1,n2);
Node right=getLowestAncestor(head.right,n1,n2);
/*左和右都不为null---说明在左子树中发现过n1或n2,
* 在右子树上也发现过n1或n2,并且n1和n2在当前节点首次相遇
*/
if(left!=null && right!=null){
return head;
}
/*左和右中一个不为null,另一个为null,
* 说明不为null的节点是n1或n2中的一个,或者是n1和n2的最近祖先;
* 直接返回;
*/
if(left!=null){
return left;
}
if(right!=null){
return right;
}
//左和右均为null,没有发现n1和n2;
return null;
}
}
递归理解
今天刚看到这个题目的递归解法时,感觉很绕,觉得方法很好,可就是想不清楚递归遍历到树的更深层级的返回值应该是什么。实在没办法理解,拿笔画了画,终于get到点了,就有了下一小结的递归执行过程图。