一、双重遍历法(第一次解法)
思路:从根结点root开始,判断该结点是否同时为p,q的祖先,并同时判断root的左右结点,是否满足为同时p,q的结点;如果在递归过程中找到root为p,q祖先结点但是其子节点不满足,则说明root为最近公共祖先;
class Solution {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if (root == null){
return null;
}
boolean isFather = isChild(root,p)&&isChild(root,q);
boolean isFatherLeft = isChild(root.left,p)&&isChild(root.left,q);
boolean isFatherRight = isChild(root.right,p)&&isChild(root.right,q);
if (isFather && !(isFatherLeft || isFatherRight)){
return root;
}else{
if (root.left!=null&&isFatherLeft){
return lowestCommonAncestor(root.left,p,q);
}
else{
return lowestCommonAncestor(root.right,p,q);
}
}
}
// 判断child结点是否为node结点的子节点
public boolean isChild(TreeNode node,TreeNode child){
if (node == null){
return false;
}
// 如果两结点相等
if (child.val == node.val) {
return true;
}else{
return isChild(node.left,child)||isChild(node.right,child);
}
}
}
结果:思路没有问题,通过了全部用例,但是超时了。。。
分析:每次判断一个结点是否为另一个结点的子节点时,都需要遍历整棵二叉树一次,时间复杂度为o(n);递归root结点时还需要遍历一次二叉树,因此总的时间复杂度至少为o(n^2)(且一次递归中需要进行多次是否为子节点的判断)
二、后序递归法(参考解法)
思路:从根节点root开始,进行后序遍历,目的是分别找到等于p,q的结点。若找到p,q结点,将其保存下来,记为left和right,进行返回。当遍历结果为空时,同样返回。对于找到的left和right,如果都不为空,则直接返回root(即root结点的左右子树分别为p,q,则root一定为为最近公共祖先);如果有一个为空,另一个不为空,则返回不为空的结果(找到了),没找到的在后序递归中继续寻找。若均为空,则返回空,同样等待继续寻找。
class Solution {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if (root == null||root.val == p.val||root.val == q.val){
return root;
}
// 递归遍历,寻找p,q
TreeNode left = lowestCommonAncestor(root.left,p,q);
TreeNode right = lowestCommonAncestor(root.right,p,q);
// 逻辑处理
if (left!=null&&right!=null){
return root;
}else{
return left!=null?left:right;
}
}
}
备注:
- 递归过程中只找到一个就返回的原因:即使这颗树的子树有还另一个(p/q),最近公共祖先仍为先找到的结点
- 总体的流程是递归+回溯
总结
刷题时要注意解法的时间复杂度。对于二叉树的题目,基本都是0(n)的时间复杂度解决(即一次遍历)