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

最新推荐文章于 2024-09-27 15:30:00 发布

如花美眷似水流年Nancy

最新推荐文章于 2024-09-27 15:30:00 发布

阅读量665

点赞数

分类专栏：保研刷题文章标签： leetcode c++ 深度优先

本文链接：https://blog.csdn.net/Nancy9669/article/details/124106364

版权

二叉树深度优先搜索最低公共祖先 DFS 算法

关键词由CSDN通过智能技术生成

保研刷题专栏收录该内容

39 篇文章 1 订阅

订阅专栏

与树相关的题大部分会用到dfs

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void dfs(TreeNode* root,TreeNode* target, vector<TreeNode*> &ancestorList, int *flag)
    {
        if(!root) return;
        if(root==target)
        {
            ancestorList.push_back(target);
            *flag=1;
            return;
        }
        ancestorList.push_back(root);
        dfs(root->left,target,ancestorList,flag);
        if(*flag==1) return;
        dfs(root->right,target,ancestorList,flag);
        if(*flag==1) return;
        ancestorList.pop_back();
    }
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        vector<TreeNode*> r1,r2;
        int flag=0;
        dfs(root,p,r1,&flag);
        flag=0;
        dfs(root,q,r2,&flag);
        int p1=0;
        while(p1<r1.size()&&p1<r2.size()&&r1[p1]==r2[p1])
            p1++;
        return r1[p1-1];
    }
};