每日力扣-树-普通二叉树最近祖先

93 阅读1分钟

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

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

解题思路:两种情况:1.p q节点在不同子树,那么最近那个分岔路口就是共同祖先。2.q或p是另外一个节点的祖先 采用后续遍历,首先默认所有节点都是false,然后遍历叶子节点,把p/q两个节点染成true,再把所有子节点存在true的节点染成true,如果一个节点他的左右节点都是true,或者节点本身是p/q,子节点存在true,那他就是最近祖先

class Solution {
public:
    vector<int> qval;
    vector<int> pval;
    TreeNode* ans;
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        dfs(root,p,q);
        return ans;
    }
    bool dfs(TreeNode* root,TreeNode* p , TreeNode* q)
    {
        if(root == nullptr) return false;
        bool l = dfs(root->left,p,q);
        bool r = dfs(root->right,p,q);
        if((l&&r) || ((root->val == p->val || root->val == q->val)&&(l || r)))
        {
            ans = root;
        }
        return l || r || (root->val == q->val || root->val == p->val);
    }
};