方法一:深度优先搜索
如果两个二叉树都为空,则两个二叉树相同
如果两个二叉树中有且只有一个为空,则两个二叉树一定不相同
如果两个二叉树都不为空,那么首先判断它们的根节点的值是否相同,若不相同则两个二叉树一定不同,若相同,再分别判断两个二叉树的左子树是否相同以及右子树是否相同
这是一个递归的过程,因此可以使用深度优先搜索,递归地判断两个二叉树是否相同
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 || p != nullptr && q == nullptr) return false;
else if (p->val != q->val) return false;
else
{
return isSameTree(p->left, q->left) && isSameTree(p->right, q->right);
}
}
};
int main()
{
Solution A;
TreeNode* t1 = new TreeNode(1, new TreeNode(2, nullptr, nullptr), new TreeNode(3, nullptr, nullptr));
TreeNode* t2 = new TreeNode(1, new TreeNode(2, nullptr, nullptr), new TreeNode(3, nullptr, nullptr));
cout << A.isSameTree(t1, t2) << endl;
return 0;
}
时间复杂度:O(min(m,n)),其中 m 和 n 分别是两个二叉树的节点数。对两个二叉树同时进行深度优先搜索,只有当两个二叉树中的对应节点都不为空时才会访问到该节点,因此被访问到的节点数不会超过较小的二叉树的节点数。
空间复杂度:O(min(m,n)),其中 m 和 n 分别是两个二叉树的节点数。空间复杂度取决于递归调用的层数,递归调用的层数不会超过较小的二叉树的最大高度,最坏情况下,二叉树的高度等于节点数
方法二:广度优先搜索
使用两个队列分别存储两个二叉树的节点。初始时将两个二叉树的根节点分别加入两个队列。每次从两个队列各取出一个节点,进行如下比较操作:
①比较两个节点的值,如果两个节点的值不相同则两个二叉树一定不同;
②如果两个节点的值相同,则判断两个节点的子节点是否为空,如果只有一个节点的左子节点为空,或者只有一个节点的右子节点为空,则两个二叉树的结构不同,因此两个二叉树一定不同
③如果两个节点的子节点的结构相同,则将两个节点的非空子节点分别加入两个队列,子节点加入队列时需要注意顺序,如果左右子节点都不为空,则先加入左子节点,后加入右子节点
④如果搜索结束时两个队列同时为空,则两个二叉树相同。如果只有一个队列为空,则两个二叉树的结构不同,因此两个二叉树不同
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
{
queue<TreeNode*> q1;
queue<TreeNode*> q2;
q1.push(p); q2.push(q);
while (!q1.empty() && !q2.empty())
{
TreeNode* tmp1 = q1.front(); q1.pop();
TreeNode* tmp2 = q2.front(); q2.pop();
if (tmp1->val != tmp2->val)
{
return false;
}
else if (tmp1->left == nullptr && tmp1->right == nullptr && tmp2->left == nullptr && tmp2->right ==nullptr)
{
continue;
}
else if (tmp1->left == nullptr && tmp2->left != nullptr)
{
return false;
}
else if (tmp1->right == nullptr && tmp2->right != nullptr)
{
return false;
}
else if (tmp1->left != nullptr && tmp2->left == nullptr)
{
return false;
}
else if (tmp1->right != nullptr && tmp2->right == nullptr)
{
return false;
}
else
{
if (tmp1->left != nullptr)
{
q1.push(tmp1->left);
q2.push(tmp2->left);
}
if (tmp1->right != nullptr)
{
q1.push(tmp1->right);
q2.push(tmp2->right);
}
}
}
return q1.empty() && q2.empty();
}
}
};
int main()
{
Solution A;
/*TreeNode* t1 = new TreeNode(1, new TreeNode(2, nullptr, nullptr), new TreeNode(3, nullptr, nullptr));
TreeNode* t2 = new TreeNode(1, new TreeNode(2, nullptr, nullptr), new TreeNode(2, nullptr, nullptr));*/
TreeNode* t1 = new TreeNode(1, nullptr, nullptr);
TreeNode* t2 = new TreeNode(1, nullptr, new TreeNode(2, nullptr, nullptr));
cout << A.isSameTree(t1, t2) << endl;
return 0;
}
时间复杂度:O(min(m,n)),其中 m 和 n 分别是两个二叉树的节点数。对两个二叉树同时进行广度优先搜索,只有当两个二叉树中的对应节点都不为空时才会访问到该节点,因此被访问到的节点数不会超过较小的二叉树的节点数
空间复杂度:O(min(m,n)),其中 m 和 n 分别是两个二叉树的节点数。空间复杂度取决于队列中的元素个数,队列中的元素个数不会超过较小的二叉树的节点数