236. 二叉树的最近公共祖先（Medium）

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

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

百度百科中最近公共祖先的定义为：“对于有根树 T 的两个节点 p、q，最近公共祖先表示为一个节点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”

示例 1：

输入：root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
输出：3

解释：节点 5 和节点 1 的最近公共祖先是节点 3 。

思路（dfs）

思考什么情况下为公共的祖先节点

• 该节点左右子树各含有目标节点
• 该节点本身即为某目标节点，其某方向的子树含另一个目标节点

if ((left && right) || ((root->val == p->val || root->val == q->val) && (left || right))) {
            ancestor = root;
}

那么上面的即为判断条件了，我们用bool值来标记各个孩子节点的情况

 bool left = dfs(root->left, p, q);// 递归左子树，判断其真值情况
 bool right = dfs(root->right, p, q);// 递归右子树，判断其真值情况

从下往上递归，上一层的子节点的bool值由下一层的节点的情况决定

图片来自力扣官方题解
https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree/solution/

源代码

class Solution {
public:
    TreeNode* ancestor;
    bool dfs(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == nullptr) {
            return false;
        }
        bool left = dfs(root->left, p, q);// 递归左子树，判断其真值情况
        bool right = dfs(root->right, p, q);// 递归右子树，判断其真值情况
        if ( (left && right) || ((root->val == p->val || root->val == q->val) && (left || right))) { // 如果该节点两侧都有其目标节点，或者改节点本身为目标节点，其某一方向子树含另一节点，则改节点为最近的祖先节点
            ancestor = root;// 因为是从下往上的，故第一个发现的深度最深，为最近的祖先节点
        }
        return left || right || (root->val == p->val || root->val == q->val);// 按照情况返回真值
    }

    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
       dfs(root,p,q);
       return ancestor;
    }
};