剑指 Offer 68 - II. 二叉树的最近公共祖先
题目描述
解题思路
还是分几种情况,其中一个等于根,一个左边一个右边,全在左边,全在右边
/*
* 剑指 Offer 68 - II. 二叉树的最近公共祖先
* 2020/8/1
* 三种情况:
* 1、p q 一个在左子树 一个在右子树 那么当前节点即是最近公共祖先
* 2、p q 都在左子树
* 3、p q 都在右子树
*/
public TreeNode lowestCommonAncestor2(TreeNode root, TreeNode p, TreeNode q) {
if(root == null) {
return null;
}
if(root == p || root == q) {
return root;
}
TreeNode left = lowestCommonAncestor(root.left, p, q);
TreeNode right = lowestCommonAncestor(root.right, p, q);
if(left != null && right != null) { //如果一个在左边一个在右边
return root;
}
if(left == null) { //如果全在左边
return right;
}
if(right == null) { //如果全在左边
return left;
}
return null;
}