题目描述
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 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 为不同节点且均存在于给定的二叉树中。
递归法
找最近简单子问题:在当前节点的左右子树中,分别查询两个节点p、q,如果左子树查找结果为空,则返回右子树查找结果;如果右子树查找结果为空,则返回左子树查找结果;如果左右子树都返回节点,则返回当前根节点。结束条件为:当前节点为空或者当前节点为p、q。
解题步骤:
1、递归查询两个节点p、q,如果某个节点等于节点p或节点q,则返回该节点的值给父节点。
2、如果当前节点有一个子树的返回值为p或q节点,则返回该值。(即告诉父节点有一个节点存在其子树中)
如果当前节点的两个子树返回值都为空,则返回空指针。
如果当前节点的左右子树分别包括p和q节点,那么当前节点为所求的解。
边界条件:
1、空树;
2、p为根结点root;
3、q为根结点root。
递归代码模板:
递归结束条件、处理当前层逻辑、下沉到下一层、清理当前层(非必须)。
具体代码实现:
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(!root || root == p || root == q) return root; //结束条件
auto leftFather = lowestCommonAncestor(root->left,p,q); //下沉到下一层
auto rightFather = lowestCommonAncestor(root->right,p,q); //下沉到下一层
if(!leftFather) return rightFather; //当前层逻辑
if(!rightFather) return leftFather; //当前层逻辑
return root;
}
};
时间复杂度分析:
因为需要对每个节点遍历一次(最差情况下),因此时间复杂度为O(n).
AC结果: