上一题,二叉搜索树的最近公共祖先,由于二叉搜索树是有序的,而二叉树是无序的,所以解法会不同。
1.分析
后序遍历(左、右、根)dfs
最近公共祖先有三种情况:
1.p,q分别在root的左右子树中
2.p=root,且 q在 root的左或右子树中;
3.q = root,且 p 在 root 的左或右子树中;
递归终止条件:
1.root为空,返回null
2.root=p或root=q,返回root
递归左子树root.left,
递归右子树root.right。
递归结果:
1.如果递归左右子树都为空,返回空
2.如果都不为空,说明root就是最近公共祖先,返回root
3.如果左为空且右不为空,p,q 都不在 root的左子树中,直接返回right
有两种情况
①q其中一个在 root的 右子树 中,此时 right指向p(假设为 p );
②p,q 两节点都在 root的 右子树 中,此时的right 指向 最近公共祖先节点
4.分析同3
2.代码(自底向上)
class Solution {
public TreeNode lowestCommonAncestor(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)//对左子树和右子树都没有公共父亲,说明root左右子树都不包含??
return null;
else if(left!=null&&right!=null)//说明pq在root异侧
return root;
else if(left==null&&right!=null)//都在右
return right;
else //都在左
return left;
}
}
3.复杂度
时间复杂度 O(N) : 其中 N 为二叉树节点数;最差情况下,需要递归遍历树的所有节点。
空间复杂度 O(N) : 最差情况下,递归深度达到 N ,系统使用O(N) 大小的额外空间。