【每日一題】100.相同的樹

題目:https://leetcode-cn.com/problems/same-tree/

解法 1,遞歸
很容易理解。

class Solution {
public:
    bool isSameTree(TreeNode* p, TreeNode* q) {
        if (p == nullptr && q == nullptr) return true;
        if (p == nullptr || q == nullptr) return false;

        return (p->val == q->val)
            && (isSameTree(p->left, q->left))
            && (isSameTree(p->right, q->right));
    }
};

解法 2,非遞歸
非遞歸方法類似於將快排的非遞歸寫法。

class Solution {
public:
    bool isSameTree(TreeNode* p, TreeNode* q) {
        if (p == nullptr && q == nullptr) return true;
        if (p == nullptr || q == nullptr) return false;

        stack<map<char, TreeNode*>> s;
        map<char, TreeNode*> m1;
        m1['p'] = p;  m1['q'] = q;
        s.push(m1);

        while (!s.empty()) {
            auto m = s.top();  s.pop();
            if (m['p'] == nullptr && m['q']==nullptr) continue;
            if ((m['p'] != nullptr && m['q'] == nullptr)
                || (m['p'] == nullptr && m['q'] != nullptr)) 
                return false;
            if (m['p']->val != m['q']->val) return false;
            
            map<char, TreeNode*> tmp;
            tmp['p'] = m['p']->left;  tmp['q'] = m['q']->left;
            s.push(tmp);
            tmp['p'] = m['p']->right;  tmp['q'] = m['q']->right;
            s.push(tmp);
        }
        return true;
    }
};

EOF

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