LeetCode236. 二叉樹的最近公共祖先

題目:

給定一個二叉樹, 找到該樹中兩個指定節點的最近公共祖先。百度百科中最近公共祖先的定義爲:“對於有根樹 T 的兩個結點 p、q,最近公共祖先表示爲一個結點 x,滿足 x 是 p、q 的祖先且 x 的深度儘可能大(一個節點也可以是它自己的祖先)。”

思路:

1. 遞歸解法:(樹這種結構真的太適合遞歸解法了):如果兩個節點分別在左右子樹,那麼公共祖先爲根結點,如果兩個節點同在一邊子樹,那麼公共祖先必在該邊子樹。
遞歸解法看起來簡單,但是仔細理解起來還是不容易,設計時候也容易出錯。可以把遞歸看成是自己調用另一個和自己功能一樣,但是函數名不同的函數來理解

2.非遞歸解法:
 

題解:(C++)

1.遞歸解法:

struct TreeNode* lowestCommonAncestor(struct TreeNode* root, struct TreeNode* p, struct TreeNode* q) {
    if(root == nullptr || root == p || root == q) return root; //遞歸終止條件
    TreeNode* left = lowestCommonAncestor(root->left,p,q); 
    TreeNode* right = lowestCommonAncestor(root->right,p,q);
    //看返回哪一個
    return left == nullptr?right:(right == nullptr?left:root); //左邊也有右邊也有那就爲root,不可能left和right返回都爲空
}

非遞歸解法:

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == NULL)
            return NULL;
        stack<TreeNode*> record;
        stack<TreeNode*> help;
        record.push(root);
        while (!record.empty())
        {
            TreeNode* cur = record.top();
            help.push(cur);
            record.pop();

            if (cur->left != NULL)
                record.push(cur->left);
            if (cur->right != NULL)
                record.push(cur->right);
        }
        while (!help.empty())
        {
            TreeNode* node = help.top();

            if (node->left != NULL && node->left->val == p->val)
                p->val = node->val;
            if (node->right != NULL && node->right->val == p->val)
                p->val = node->val;
            if (node->left != NULL && node->left->val == q->val)
                q->val = node->val;
            if (node->right!= NULL && node->right->val == q->val)
                q->val = node->val;
            if (p->val == q->val)
                return p;
            help.pop();
        }
        return NULL;
    }
};

知識點:

nullptr 爲空指針

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章