给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
示例:
输入:root=[3,5,1,6,2,0,8,null,null,7,4],p=5,q=1
输出:3
解释:节点1和节点5的最近公共祖先是节点3.
class Solution {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if (root == p || root == q || root == null) return root;
TreeNode left = lowestCommonAncestor(root.left, p, q);
TreeNode right = lowestCommonAncestor(root.right, p, q);
if (left != null && right != null) return root;
if (left == null) return right;
return left;
}
}
题解:root:当前树的根节点;p:要寻找的第一个节点;q:要寻找的第二个节点。返回值类型是TreeNode,表示找到的最近公共祖先节点。
如果当前节点是否为p或q,或者是否为空节点。则返回root节点,因为我们找到了其中一个节点。否则返回null,表示没有找到任何节点。
递归调用分别搜索左子树和右子树,结果存储在left和right中。
如果left和right都不为null,这意味p和q分别在root的左右子树中,因此当前节点是它们的最近公共祖先。
如果left为null,说明p和q都不在左子树中,那么返回right,即右子树中的结果。
如果right为null,说明p和q都不在右子树中,那么返回left,即左子树中的结果。
时间复杂度为O(N),其中N是节点数,空间复杂度为O(H),H是树的高度。