给定一个二叉树, 找到该树中两个指定节点的最近公共祖先
百度百科中最近公共祖先的定义为:对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。
解题思路:
从某个节点出发,如果能同时找到 p 和 q,说明这个节点是公共祖先。那么如何判断最近的公共祖先呢?
从该节点出发进行查找时,p 和 q 主要从三个渠道进行查找:
- 该节点自身是不是 p 或 q
- 该节点的左子树是否包含 q 或 q
- 该节点的右子树是否包含 p 或 q
如果 p 和 q 来自于这三个渠道中的两个,那么这个节点就是最近公共祖先
// 为了更方便的获取到最终结果,使用一个成员变量来记录,最近公共祖先 lca
public TreeNode lca = null;
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if (root == null) {
return null;
}
// 递归查找 p 和 q,用一个辅助方法进行递归
findNode(root, p, q);
return lca;
}
// 如果当前 root 中能找到 p 或者 q,找到一个就返回 true
// 如果 p 和 q 都没找到,就返回 false
public boolean findNode(TreeNode root, TreeNode p, TreeNode q) {
if (root == null) {
// 空树里肯定找不到,就返回 false
return false;
}
// 使用 ? : 把 boolean 转换成整数
// 从三个渠道分别查找
// 判定根结点是不是 p 或 q
int mid = (root == p || root == q) ? 1 : 0;
// 在左子树中查找 p 和 q
int left = findNode(root.left, p, q) ? 1 : 0;
// 在右子树中查找 p 和 q
int right = findNode(root.right, p, q) ? 1 : 0;
// 最重要的步骤,判定该结点是否是 lca
// 如果 p 和 q 来自于这三个渠道中的两个,那么这个结点就是 lca
if (mid + left + right == 2) {
lca = root;
}
// 只要找到了其中的 p 或者 q 中的一个, 都可以返回 true
// 要是找到两个也是返回 true
// 一个都没找到才是返回 false
return (left + right + mid) > 0;
}