题目
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。最近公共祖先的定义为:“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
示例1
输入:root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
输出:3
解释:节点 5 和节点 1 的最近公共祖先是节点 3 。
示例2
输入:root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
输出:5
解释:节点 5 和节点 4 的最近公共祖先是节点 5 。因为根据定义最近公共祖先节点可以为节点本身。
- 树中节点数目在范围 [2, 105] 内
- -109 <= Node.val <= 109
- 所有 Node.val 互不相同
- p != q
- p 和 q 均存在于给定的二叉树中
分析
找出两个节点的最近公共节点,有两种解决方式,一种是从两个要查找的节点到根节点上的路径都连接起来,那么这两条路径就相当于两个链表了,这题就变成了查找两个链表的第一个公共节点了。
可以遍历这棵二叉树,然后使用一个 map 记录所遍历节点的父节点,然后在根据当前节点一直往上查找父节点,一直到根节点,这种方式比较简单,我们再来看另一种方式。
查找两个节点的最近公共节点,也就是从下往上找,我们知道二叉树都是从上往下遍历的,没法从下往上遍历。也就是说如果知道一个节点,肯定能找到他的子节点,但是我们没法找到他的父节点。
这里再来回顾一下递归,对于一棵二叉树来说因为是从根节点开始的,当递归往回走的时候就相当于从下往上遍历,所以本题可以参考二叉树的后序遍历来写。
1,如果两个节点都在左子树上,就返回左子树节点。
2,如果两个节点都在右子树上,就返回右子树节点。
3,如果两个节点分别在两棵子树上,说明当前节点就是他俩的最近公共祖先节点,直接返回当前节点即可。
代码
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if (root == null || p == root || q == root)
return root;
// 参考二叉树的后序遍历
TreeNode left = lowestCommonAncestor(root.left, p, q);
TreeNode right = lowestCommonAncestor(root.right, p, q);
if (left == null)// 左子树为空,肯定都在右子树上
return right;
if (right == null)// 右子树为空,肯定都在左子树上
return left;
// 左右子树都不为空,一个在左子树一个在右子树,所以root就是他们的最近公共祖先节点。
return root;
}
原文:https://mp.weixin.qq.com/s/v-ZYi8MLXZj7QX5GvuNq3A