题目给出了p,q都是树里的节点。那么假如第二层的左节点是p,与它同层的节点(即二叉树第二层右节点)没找到q,则q一定在p的子树里,因为q是树里的节点。最近公共祖先就是p。
三种情况:
- p和 q 在 root 的子树中,且分列 root 的 异侧(即分别在左、右子树中)
- p = root,且 q 在 root 的左或右子树中
- q = root,且 p在 root 的左或右子树中
通过后序递归查找左右子树存在下述情况:
- 如果left和right都为null,说明对于调用递归方法的树root,它的左右子树里p和q一个都没找到,直接返回null。
- 如果left和right都不为null,说明p和q分别在root的两侧,root就是它们的最近公共祖先。
- 如果right为空,left不为空,说明
- 若此时root是根节点,则p,q都在左子树里
- 若此时p或q是某一层的root,则当前root是另一个的祖先
- 如果left为空,而right不为空,说明是与情况3相反的情况。
遇到一个p/q,然后直接向上返回这个p/q(不是p/q都返回null)。如果在某个节点遇到另一子树也返回上来了p/q(即两子树返回值均不为空),这个节点就是最近公共祖先。一直遇不到的话这个被一直向上返回的p/q就最终被返回(另一个没有被发现的q/p是这个节点的后代)。
l(3)代表递归函数中正在递归节点3
l代表递归左子树的返回值(即left)
r代表递归右子树的返回值(即right)
public static TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q){
if(root == null || 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 null;
if(left == null) return right;
if(right == null) return left;
return root;// 2. if(left != null and right != null)
}
示例2: