码迷,mamicode.com
首页 > 其他好文 > 详细

【LeetCode】Same Tree (2 solutions)

时间:2014-12-01 23:49:09      阅读:192      评论:0      收藏:0      [点我收藏+]

标签:des   style   blog   http   io   ar   color   sp   for   

Same Tree

Given two binary trees, write a function to check if they are equal or not.

Two binary trees are considered equal if they are structurally identical and the nodes have the same value.

 

解法一:递归

/**
 * Definition for binary tree
 * 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) {
        if(!p && !q)
            return true;
        else if(!p && q)
            return false;
        else if(p && !q)
            return false;
        else
        {
            if(p->val != q->val)
                return false;
            else
                return isSameTree(p->left, q->left) && isSameTree(p->right, q->right);
        }
    }
};

bubuko.com,布布扣

 

解法二:非递归

建立两个队列分别进行层次遍历,进队时检查对应点是否相等

/**
 * Definition for binary tree
 * 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) {
        if(!p && !q)
            return true;
        else if(!p && q)
            return false;
        else if(p && !q)
            return false;
        else
        {
            if(p->val != q->val)
                return false;
            else
            {    
                queue<TreeNode*> lq;
                queue<TreeNode*> rq;
                lq.push(p);
                rq.push(q);
                while(!lq.empty() && !rq.empty())
                {
                    TreeNode* lfront = lq.front();
                    TreeNode* rfront = rq.front();
                    lq.pop();
                    rq.pop();
                    if(!lfront->left && !rfront->left)
                        ;// null
                    else if(!lfront->left && rfront->left)
                        return false;
                    else if(lfront->left && !rfront->left)
                        return false;
                    else
                    {
                        if(lfront->left->val != rfront->left->val)
                            return false;
                        else
                        {
                            lq.push(lfront->left);
                            rq.push(rfront->left);
                        }
                    }
                    if(!lfront->right && !rfront->right)
                        ;// null
                    else if(!lfront->right && rfront->right)
                        return false;
                    else if(lfront->right && !rfront->right)
                        return false;
                    else
                    {
                        if(lfront->right->val != rfront->right->val)
                            return false;
                        else
                        {
                            lq.push(lfront->right);
                            rq.push(rfront->right);
                        }
                    }    
                }
                return true;
            }
        }
    }
};

bubuko.com,布布扣

【LeetCode】Same Tree (2 solutions)

标签:des   style   blog   http   io   ar   color   sp   for   

原文地址:http://www.cnblogs.com/ganganloveu/p/4136259.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!