No.119 - LeetCode236 - (经典) - 两个节点的最近公共祖先

原创于 2019-09-24 23:14:02 发布 · 137 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode

leetcode 专栏收录该内容

251 篇文章

订阅专栏

本文介绍了一种使用递归回溯方法寻找二叉树中两个节点的最低公共祖先的有效算法。通过定义一个解决方案类，该算法利用两个布尔变量跟踪目标节点，实现了一次遍历确定公共祖先。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

用两个变量，回溯时判断即可。

/**
 * 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:
    TreeNode* ans;
    void dfs(TreeNode* now,TreeNode* p,TreeNode* q,bool& flagP,bool& flagQ){
        bool A = false;
        bool B = false;
        if(now->left != NULL) dfs(now->left,p,q,A,B);
        if(now->right != NULL) dfs(now->right,p,q,A,B);
        if(now == p) A = true;
        if(now == q) B = true;
        if(ans == NULL && A && B) ans = now;
        flagP = flagP || A;
        flagQ = flagQ || B;
    }
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        bool flagP = false;
        bool flagQ = false;
        ans = NULL;
        dfs(root,p,q,flagP,flagQ);
        if(ans == NULL && flagP == flagQ) ans = root;
        return ans;
    }
};