给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4]
示例 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。
因为根据定义最近公共祖先节点可以为节点本身。
说明:
- 所有节点的值都是唯一的。
- p、q 为不同节点且均存在于给定的二叉树中。
思路:利用二叉树的先序遍历,作为最深的公共祖先节点,两个节点必定分别分布在二叉树的左右子树中,因此只需在遍历时进行记录,左子树和右子树中都搜索到了,那么该节点一定是最深的公共祖先节点;否则,他可能是最深公共祖先的祖先节点(两个节点都在同一边),则返回它即可;否则没找到返回null。
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if(root == p || root == q || root == null)
return root;
TreeNode leftNode = lowestCommonAncestor(root.left, p, q);
TreeNode righNode = lowestCommonAncestor(root.right, p, q);
if(leftNode != null && righNode != null)
return root;
else
return leftNode != null? leftNode: righNode;
}