二叉树的最近公共祖先
题目描述
例如,给定如下二叉树: 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 为不同节点且均存在于给定的二叉树中。
题目分析
这题Kotlin那部分有Bug,提交上去报错的,思路完全一样代码也怎么变的Java代码完美运行!
思路:分两种情况
- 当前节点和p或者q相同,返回当前节点就行
- 当前节点和p且q都不相同,就必须在左子树和右子树里找到p和q
这里又分两种情况:
- p和q都在左子树,那么pq的最近公共祖先就在左子树,回到(1)
- p和q都在右子树,那么pq的最近公共祖先就在右子树(1)
- p和q一边一个,那么当前节点就是最近公共祖先,返回当前节点
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if(root == null || root.val == q.val || root.val == p.val)
return root;
TreeNode leftNode = lowestCommonAncestor(root.left,p,q);
TreeNode rightNode = lowestCommonAncestor(root.right,p,q);
if(leftNode == null)
return rightNode;
if(rightNode == null)
return leftNode;
return root;
}