二叉树的最近公共祖先

给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为:“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”

最近公共祖先意味着需要从下往上遍历,这就是后序遍历(左右中)的地盘了。

寻找最近祖先(以下这套逻辑包含了两种情况,情况一:p、q都为叶子节点,情况二:p、q有一个为叶子节点,另一个为根节点):

  • 入口函数:如果节点为p或者q则返回
  • 中间节点逻辑处理:
    • 如果left和right有返回值,则root为最近祖先
    • 如果left有返回值,则将left返回
    • 如果right有返回值,则将right返回
    • 如果left和right为空,则返回空

代码如下:


class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root==p || root==q || root==nullptr) return root;
        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        TreeNode* right = lowestCommonAncestor(root->right, p, q);
        if (left && right) return root;
        else if (left && right==nullptr) return left;
        else if (right && left==nullptr) return right;
        else return nullptr;
    }
};