这道题目与堆成二叉树题目很相似,对称二叉树是每次传入的节点组合是 《左子树的左节点 和 右子树的 右节点》 《左子树的右节点 和 右子树的左节点》, 而这道题就更加简单,传入两棵树的左节点和右节点即可。
/*** Definition for a binary tree node.* struct TreeNode {* int val;* TreeNode *left;* TreeNode *right;* TreeNode() : val(0), left(nullptr), right(nullptr) {}* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}* };*/
class Solution {
public:bool isSameTree(TreeNode* p, TreeNode* q) {if(p == nullptr && q == nullptr){return true;}else if(p == nullptr || q == nullptr){return false;}else if(p->val != q->val){return false;}bool l_tree = isSameTree(p->left, q->left);bool r_tree = isSameTree(p->right, q->right);return l_tree && r_tree;}
};
尝试使用迭代法来完成这道题目,通过队列每次出队两个元素,并对其进行比较,从而判断两个树是否相等。
尽管代码思路很简单,但在实现时,一开始习惯性的写成了层序遍历那样,在将子节点入队时,判断了子节点不为空才让子节点入队,导致出现误判的情况。这里即使是空节点,也是需要入队进行比较的,因为一个树的这个节点为空,但另一棵树对应的这个节点可不一定。
class Solution {
public:bool isSameTree(TreeNode* p, TreeNode* q) {queue<TreeNode*> que;if(p != nullptr && q != nullptr){que.push(p);que.push(q);}else if(p == nullptr && q == nullptr){return true;}else{return false;}while(!que.empty()){TreeNode* node1 = que.front();que.pop();TreeNode* node2 = que.front();que.pop();if(node1 == nullptr && node2 == nullptr){continue;}if(node1 == nullptr || node2 == nullptr){return false;}if(node1->val != node2->val){return false;}que.push(node1->left);que.push(node2->left);que.push(node1->right);que.push(node2->right);}return true;}
};