题目描述:
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 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 。因为根据定义最近公共祖先节点可以为节点本身。
示例 3:
输入:root = [1,2], p = 1, q = 2
输出:1
提示:
树中节点数目在范围 [2, 105] 内。
-109 <= Node.val <= 109
所有 Node.val 互不相同 。
p != q
p 和 q 均存在于给定的二叉树中。
解题思路:
使用递归、回溯的思想,自顶向下的层层递进,回溯的时候自底向上所以我们最终得到的一定是最近的公共祖先。
详见代码详细注释。
代码:
class Solution {
//自顶向下搜索,若一个节点的左右子树中包含p、q节点,那么先满足条件的节点就是最近公共祖先,因为回溯的时候时自底向上的
//时间复杂度O(N)
//空间复杂度O(N)
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
//查找结束,查找到叶子节点或者p、q节点就代表着查找结束
if(root == null || root == p || root == q){
return root;
}
//去查找p、q节点是否在该节点的左右子树中
TreeNode left = lowestCommonAncestor(root.left,p,q);
TreeNode right = lowestCommonAncestor(root.right,p,q);
//两端都查不到p、q
if(left == null && right == null){
return null;
}
//p、q两个节点,在当前节点的左右两个子树中
else if(left != null && right != null){
return root;
}
//p、q两个节点在当前节点的左子树或者右子树中
else{
return (left == null) ? right : left;
}
}
}