1. 程式人生 > >LeetCode領釦 #100 相同的樹(Same Tree)

LeetCode領釦 #100 相同的樹(Same Tree)

給定兩個二叉樹,編寫一個函式來檢驗它們是否相同。

如果兩個樹在結構上相同,並且節點具有相同的值,則認為它們是相同的。

示例 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;		    
    }
};