给你两棵二叉树的根节点 p
和 q
,编写一个函数来检验这两棵树是否相同。
如果两个树在结构上相同,并且节点具有相同的值,则认为它们是相同的。
示例 1:
输入:p = [1,2,3], q = [1,2,3] 输出:true
示例 2:
输入:p = [1,2], q = [1,null,2] 输出:false
示例 3:
输入:p = [1,2,1], q = [1,1,2] 输出:false
解法一:深度优先遍历
/**
* 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;
// 如果只有一个树是空的,则它们不相同
if (p == nullptr || q == nullptr) return false;
// 如果两个树的根节点的值不相同,则它们不相同
if (p->val != q->val) return false;
// 递归地检查左子树和右子树是否相同
return isSameTree(p->left, q->left) && isSameTree(p->right, q->right);
}
};
代码中使用了递归的方法来解决这个问题。首先,判断两棵树是否都为空,如果是,则它们是相同的。如果只有一棵树为空,那么它们不相同。
接下来,判断两棵树的根节点的值是否相同,如果不相同,则它们不相同。如果根节点的值相同,那么就要分别判断它们的左子树和右子树是否相同。所以用递归的方式来判断左右子树是否相同。
如果左子树和右子树都相同,则整棵树都相同。如果有一棵子树不相同,则整棵树不相同。
最后,返回判断结果。如果两棵树相同,函数返回true;如果两棵树不相同,函数返回false。
/**
* 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;
queue<TreeNode*> queue1,queue2;
queue1.push(p);
queue2.push(q);
while(!queue1.empty()&&!queue2.empty()){
auto node1=queue1.front();
auto node2=queue2.front();
queue1.pop();
queue2.pop();
if(node1->val!=node2->val)
return false;
auto left1 = node1->left, right1 = node1->right, left2 = node2->left, right2 = node2->right;
if ((left1 == nullptr) ^ (left2 == nullptr)) {
return false;
}
if ((right1 == nullptr) ^ (right2 == nullptr)) {
return false;
}
if (left1 != nullptr) {
queue1.push(left1);
}
if (right1 != nullptr) {
queue1.push(right1);
}
if (left2 != nullptr) {
queue2.push(left2);
}
if (right2 != nullptr) {
queue2.push(right2);
}
}
return queue1.empty()&&queue2.empty();
}
};
与之前的代码相比,这个代码使用了迭代的方法来解决这个问题。
代码中首先判断两棵树是否都为空,如果是,则它们是相同的;如果只有一棵树为空,那么它们不相同。
接下来,使用两个队列queue1和queue2来分别存储两棵树的节点。将两棵树的根节点分别加入队列queue1和queue2中。
然后开始循环,当队列queue1和queue2都不为空时,取出队列的头节点,并比较它们的值是否相同。如果不相同,则直接返回false。
接下来判断左子树和右子树是否符合相同的条件。如果左子树的情况与右子树的情况不同,说明两棵树不相同,直接返回false。
如果左子树和右子树的情况相同,则将它们的左孩子和右孩子分别加入到队列queue1和queue2中。
最后,如果队列queue1和queue2都为空,则说明两棵树是相同的。如果有一棵树为空而另一棵树不为空,或者有节点的值不同,那么两棵树就是不相同的。
最后,返回判断结果。如果两棵树相同,函数返回true;如果两棵树不相同,函数返回false。