目录结构
1.题目
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4]
示例:
输入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
输出: 3
解释: 节点 5 和节点 1 的最近公共祖先是节点 3。
输入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
输出: 5
解释: 节点 5 和节点 4 的最近公共祖先是节点 5。因为根据定义最近公共祖先节点可以为节点本身。
说明:
- 所有节点的值都是唯一的。
- p、q 为不同节点且均存在于给定的二叉树中。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2.题解
递归:
- 若p和q分别存在于左右子树,则当前节点为最近公共祖先;
- 若p和q均在右子树,right要么是在右子树找到的公共祖先且不等于p或q,要么等于p或q,故返回right为最近公共祖先;
- 若p和q均在左子树,left要么是在左子树找到的公共祖先且不等于p或q,要么等于p或q,故返回left为最近公共祖先。
public class Solution236 {
@Test
public void test236() {
TreeNode left = new TreeNode(2, new TreeNode(0), new TreeNode(4, new TreeNode(3), new TreeNode(5)));
TreeNode right = new TreeNode(8, new TreeNode(7), new TreeNode(9));
TreeNode root = new TreeNode(6, left, right);
System.out.println(lowestCommonAncestor1(root, left.left, left).val);
}
private TreeNode result = null;
public boolean dfs(TreeNode root, TreeNode p, TreeNode q) {
if (root == null) {
return false;
}
boolean inCurrent = root == p || root == q;
boolean inLeft = dfs(root.left, p, q);
boolean inRight = dfs(root.right, p, q);
if ((inLeft && inRight) || inCurrent) {
result = root;
}
return inCurrent || inLeft || inRight;
}
public TreeNode lowestCommonAncestor1(TreeNode root, TreeNode p, TreeNode q) {
dfs(root, p, q);
return result;
}
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if (root == p || root == q) {
return root;
}
if (root != null) {
TreeNode left = lowestCommonAncestor(root.left, p, q);
TreeNode right = lowestCommonAncestor(root.right, p, q);
if (left != null && right != null) {
return root;
} else if (left == null) {
return right;
} else {
return left;
}
}
return null;
}
}
- 时间复杂度:
- 空间复杂度: