给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
根据以上定义,若 root是p,q 的 最近公共祖先 ,则只可能为以下情况之一:
p 和 q 在 root 的子树中,且分列 root 的 异侧(即分别在左、右子树中);
p = root,且 q 在 root的左或右子树中;
q = root,且 p 在 root的左或右子树中;
递归解析:
终止条件:
当越过叶节点,则直接返回 nul ;
当 root 等于 p, q ,则直接返回 root ;
递推工作:
开启递归左子节点,返回值记为 left ;
开启递归右子节点,返回值记为 right ;
返回值: 根据 left和 right,可展开为四种情况;
1)当 left 和 right同时为空 :说明 root 的左 / 右子树中都不包含 p,q,返回 null ;
2)当 left 和 right同时不为空 :说明 p, q分列在root 的 异侧 (分别在 左 / 右子树),因此root 为最近公共祖先,返回 root;
3)当 left为空 ,right 不为空 :p,q 都不在 root的左子树中,直接返回 right 。具体可分为两种情况
3.1)p,q其中一个在 root的右子树 中,假设是p在右子树中,则此时 right 指向p;
3.2)p,q两节点都在 root的右子树 中,此时的right指向的就是最近公共祖先节点 ;
4)当 left 不为空 , right为空 :与情况 3. 同理;
具体代码如下:
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(root == nullptr || root==p || root==q)
return root;
TreeNode* left = lowestCommonAncestor(root->left,p,q);
TreeNode* right = lowestCommonAncestor(root->right,p,q);
if(left != nullptr && right != nullptr)
return root;
return left != nullptr?left:right;
}