剑指offer系列-68 II 二叉树的最近公共祖先

最新推荐文章于 2022-10-20 20:31:02 发布

scut_lrr

最新推荐文章于 2022-10-20 20:31:02 发布

阅读量101

点赞数

分类专栏：剑指offer C++

本文链接：https://blog.csdn.net/weixin_42448226/article/details/114932768

版权

C++ 同时被 2 个专栏收录

76 篇文章 0 订阅

订阅专栏

剑指offer

73 篇文章 0 订阅

订阅专栏

// 解法一：递归

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if(root==nullptr)
            return nullptr;
        if(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;
        else if(left==nullptr) 
            return right;
        else if(right==nullptr)  
            return left;
        return nullptr;  // p,q存在的话不可能执行到这里
    }
};


//解法二：存储父子关系
class Solution {
public:
    unordered_map<int,TreeNode*> fa;
    unordered_map<int,bool> vis;
    void dfs(TreeNode* root){
        if(root==nullptr) return;
        if(root->left){
            fa[root->left->val]=root;
            dfs(root->left);
        }
        if(root->right){
            fa[root->right->val]=root;
            dfs(root->right);
        }
    }

    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        fa[root->val]=nullptr;
        dfs(root);
        while(p!=nullptr){
            vis[p->val]=true;
            p=fa[p->val];
        }
        while(q!=nullptr){
            if(vis[q->val]) return q;
            q=fa[q->val];
        }
        return nullptr;
    }
};