lowest common ancestor

lca的dfs做法,含判斷p q存在性。

class Solution {
    pair<TreeNode*, int> helper(TreeNode * root, TreeNode * p, TreeNode *q) {
        if(root == NULL) return {NULL, 0};
        auto l = helper(root->left, p, q);
        if(l.first && l.second == 2) { return l;}
        auto r = helper(root->right, p,q);
        if(r.first && r.second == 2) { return r;}
        int count = l.second + r.second;
        if(root == p || root == q) { count++; return {root,count};}
        if(count == 2) {return {root, count};}
        if(l.second == 1) {return l;}
        if(r.second == 1) {return r;}
        return {NULL, 0};
    }
    
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        auto res = helper(root, p, q);
        if(res.second == 2) {return res.first;}
        return NULL;
    }
};

不含判斷:

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if(root == NULL || p == NULL || q == NULL) return NULL;
        if(root == p || root == q) return root;
        TreeNode * l = lowestCommonAncestor(root->left, p, q);
        TreeNode * r = lowestCommonAncestor(root->right, p, q);
        if(l && r) return root;
        if(l) return l;
        if(r) return r;
        return NULL;        
    }
};    
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容