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

最新推荐文章于 2023-08-25 18:30:44 发布

原创最新推荐文章于 2023-08-25 18:30:44 发布 · 149 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode

leetcode 专栏收录该内容

251 篇文章

订阅专栏

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

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

/**
 * 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;
    }
};