首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
您现在的位置: 程式碼之家 >> 編程語言 >  >> C語言 >> C++ >> 關於C++

LeetCode Same Tree:遞歸與迭代雙法判斷二叉樹相同

     編輯:關於C++

問題描述

判斷兩棵二叉樹是否完全相同(結構相同且對應節點值相同)。

遞歸法


bool isSameTree(TreeNode* p, TreeNode* q) {
    if (!p && !q) return true;
    if (!p || !q) return false;
    return p->val == q->val
        && isSameTree(p->left, q->left)
        && isSameTree(p->right, q->right);
}

迭代法(雙隊列 BFS)


bool isSameTree(TreeNode* p, TreeNode* q) {
    if (!p && !q) return true;
    if (!p || !q) return false;
    queue q1, q2;
    q1.push(p); q2.push(q);
    while (!q1.empty()) {
        TreeNode* a = q1.front(); q1.pop();
        TreeNode* b = q2.front(); q2.pop();
        if (a->val != b->val) return false;
        if (a->left && b->left) { q1.push(a->left); q2.push(b->left); }
        else if (a->left || b->left) return false;
        if (a->right && b->right) { q1.push(a->right); q2.push(b->right); }
        else if (a->right || b->right) return false;
    }
    return true;
}

複雜度

  • 時間:O(n)
  • 空間:O(n) 隊列/棧
  1. 上一篇文章:
  2. 下一篇文章:
Copyright © 程式碼之家 All Rights Reserved