LeetCode領釦 #100 相同的樹(Same Tree)
阿新 • • 發佈:2018-12-12
給定兩個二叉樹,編寫一個函式來檢驗它們是否相同。
如果兩個樹在結構上相同,並且節點具有相同的值,則認為它們是相同的。
示例 1:
輸入: 1 1 / \ / \ 2 3 2 3 [1,2,3], [1,2,3] 輸出: true
示例 2:
輸入: 1 1 / \ 2 2 [1,2], [1,null,2] 輸出: false
示例 3:
輸入: 1 1 / \ / \ 2 1 1 2 [1,2,1], [1,1,2] 輸出: false
思路:沒有使用遞迴程式,主要是為了複習一下二叉樹的層次遍歷(佇列實現)。對2顆二叉樹進行層次遍歷,NULL結點也要算進去,若發現不同立即返回false,否則演算法繼續。
/** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */ class Solution { public: bool isSameTree(TreeNode* p, TreeNode* q) { queue<TreeNode*> que1, que2; que1.push(p); que2.push(q); while( !que1.empty() && !que2.empty() ) { TreeNode *n1 = que1.front(); TreeNode *n2 = que2.front(); que1.pop(); que2.pop(); if( !n1 && !n2 ) continue; if( (n1&&!n2) || (n2&&!n1) || n1->val!=n2->val ) return false; if( n1 ) que1.push(n1->left); if( n1 ) que1.push(n1->right); if( n2 ) que2.push(n2->left); if( n2 ) que2.push(n2->right); } if( que1.empty() && que2.empty() ) return true; return false; } };