剑指 Offer 68 - II. 二叉树的最近公共祖先

最新推荐文章于 2021-02-11 16:12:40 发布

爱学习的阿强

最新推荐文章于 2021-02-11 16:12:40 发布

阅读量76

点赞数 1

分类专栏：数据结构

本文链接：https://blog.csdn.net/yangqiang1997/article/details/110135921

版权

数据结构专栏收录该内容

6 篇文章 0 订阅

订阅专栏

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

分两种情况讨论最近公共节点如何判断
在这里插入图片描述
递归写法

TreeNode* ans;
    bool dfs(TreeNode* root, TreeNode* p, TreeNode* q){
        if(root==nullptr){
            return false;
        }
        bool lson=dfs(root->left,p,q);//递归
        bool rson=dfs(root->right,p,q);
        if((lson&&rson)||((root==p||root==q)&&(lson||rson))){//记录公共祖先结点
            ans=root;
        }
        return lson||rson||(root==p||root==q);//子树是否包括p或q
    }
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        dfs(root,p,q);
        return ans;
    }

记录父节点
遍历二叉树将每个结点与父结点的关系存储进map。
从p开始遍历把所有父结点设为true，从q开始同样操作如果遇到true则是最近公共祖先节点。

class Solution {
public:
    unordered_map<int, TreeNode*> fa;
    unordered_map<int, bool> vis;
    void dfs(TreeNode* root){//制作父结点散列表
        if (root->left != nullptr) {
            fa[root->left->val] = root;
            dfs(root->left);
        }
        if (root->right != nullptr) {
            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) {//标记p所有父结点
            vis[p->val] = true;
            p = fa[p->val];
        }
        while (q != nullptr) {//寻找q与p的最近公共祖先结点
            if (vis[q->val]) return q;
            q = fa[q->val];
        }
        return nullptr;
    }
};