Code20 相同的树

题目

leetcode100. 相同的树
给定两个二叉树，编写一个函数来检验它们是否相同。
如果两个树在结构上相同，并且节点具有相同的值，则认为它们是相同的。

代码

• C代码

// C 
// 思路：深度优先，递归
struct TreeNode {
  int val;
  struct TreeNode* left;
  struct TreeNode* right;
};

#include <cstddef>
bool isSameTree(struct TreeNode* p, struct TreeNode* q) {
  while (p && q) {
    if (p->val != q->val) {
      return false;
    }
    return isSameTree(p->left, q->left) && isSameTree(p->right, q->right);
  }

  if (((NULL == p) && q) || (p && (NULL == q))) {
    return false;
  }

  return true;
}

• C++ 代码

// 思路：广度优先，队列操作
#include <queue>
using namespace std;
class Solution {
 public:
  bool isSameTree(TreeNode* p, TreeNode* q) {
    if ((nullptr == p) ^ (nullptr == q)) {
      return false;
    }
    if ((nullptr == p) && (nullptr == q)) {
      return true;
    }
    queue<TreeNode*> q_p;
    q_p.push(p);
    queue<TreeNode*> q_q;
    q_q.push(q);
    while(!q_p.empty() && !q_q.empty()){
      auto node_p = q_p.front();
      q_p.pop();
      auto node_q = q_q.front();
      q_q.pop();

      if ((nullptr == node_p) ^ (nullptr == node_q)) {
        return false;
      }
      if ((nullptr == node_p) && (nullptr == node_q)) {
        continue;
      }

      if (node_p->val != node_q->val){
        return false;
      }

      q_p.push(node_p->left);
      q_p.push(node_p->right);
      q_q.push(node_q->left);
      q_q.push(node_q->right);
    }

    return true;
  }
};