leetcode_236

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

题目描述

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

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

题目理解

首先可以分为几种情况：

1. 节点p、q分别在当前节点的左子树和右子树中 - - > 当前节点为节点p、q的最近公共祖先。
2. 节点p或q在当前节点的左子树或者右子树中，当前节点为节点p或者节点q - - > 当前节点为节点p、q的最近公共祖先。
3. 节点p和q都在当前节点的左子树或者右子树中 - - > 当前节点不是p、q的最近公共祖先

代码

class Solution {

private:
    TreeNode* ret;

public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        findNearestCommonAncestor(root, p, q);
        return ret;
    }
    int findNearestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q){
        if (root == nullptr)
            return 0;
        
        int isLeft = findNearestCommonAncestor(root->left, p, q);
        int isRight = findNearestCommonAncestor(root->right, p, q);

        if(isLeft && isRight)
            ret = root;
        if((isLeft||isRight) && (root == p || root == q))
            ret = root;
            
        return isLeft || isRight || root == p || root == q;
    }
};