Day119 | 灵神 | 二叉树 | 二叉树的最近共公共祖先-CSDN博客

Day119 | 灵神 | 二叉树 | 二叉树的最近共公共祖先

236.二叉树的最近共公共祖先

思路：

首先我们采用后序遍历

递归函数返回值是问是「最近公共祖先的候选项」。对于最外层的递归调用者来说，返回值是最近公共祖先的意思。但是，在递归过程中，返回值可能是最近公共祖先，也可能是空节点（表示子树内没找到任何有用信息）、节点 p 或者节点 q（可能成为最近公共祖先，或者用来辅助判断上面的某个节点是否为最近公共祖先）。

1.当前节点的是空节点

那就返回空

2.当前结点是p或者q，直接返回当前结点就行

比如5是p，4是q，那么我们遍历到5就直接返回5就行，因为5是5和4的最近公共祖先

3.如果左右子树都有的话，也就是pq分别在这当前结点的左右子树，那么最近公共祖先就是当前结点

4.如果只在左子树中找到了q或者p，那说明最近公共祖先肯定在左子树，返回遍历左子树的结果就行

右子树是同理的

5.当前结点的左右子树都没找到p或者q，那说明在其他的子树

返回nullptr

完整代码：

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        //1.当前结点为空或者q或者p，返回当前结点
        if(root==nullptr || root==p || root==q)
            return root;
        TreeNode *l=lowestCommonAncestor(root->left,p,q);
        TreeNode *r=lowestCommonAncestor(root->right,p,q);
        //2.左右子树中分别找到了p或者q
        if(l!=nullptr&&r!=nullptr)
            return root;
        //3.p和q全在右子树中
        else if(l==nullptr&&r!=nullptr)
            return r;
        //4.p和q全在左子树中
        else if(l!=nullptr&&r==nullptr)
            return l;
        //5.q和p不在当前子树
        else
            return nullptr;
    }
};