给你两棵二叉树的根节点 p
和 q
,编写一个函数来检验这两棵树是否相同。
如果两个树在结构上相同,并且节点具有相同的值,则认为它们是相同的。
示例 1:
输入:p = [1,2,3], q = [1,2,3] 输出:true
官方题解
/**
* 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;
} else {
return isSameTree(p->left, q->left) && isSameTree(p->right, q->right); //一起往左边找 一起往右边找
}
}
};
常规可运行代码
#include <iostream>
#include <algorithm> // 引入 std::max 函数
using namespace std;
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;
} else {
return isSameTree(p->left, q->left) && isSameTree(p->right, q->right);
}
}
};
// 辅助函数,用于创建二叉树
TreeNode* createTreeNode(int val) {
return new TreeNode(val); //接受一个整数值val并创建一个新的TreeNode对象
}
int main(){
Solution solution;
// 创建两个相同的二叉树进行测试
TreeNode* tree1 = createTreeNode(1);
tree1->left = createTreeNode(2);
tree1->right = createTreeNode(3);
TreeNode* tree2 = createTreeNode(1);
tree2->left = createTreeNode(2);
tree2->right = createTreeNode(3);
// 测试isSameTree函数
cout << "Are the two trees the same? " << (solution.isSameTree(tree1, tree2) ? "Yes" : "No") << endl;
return 0;
}
/*
p.val 访问树根
p.left p.righ 访问树杈
p == null 判断这个指针是不是空的
isSameTree(p.left, q.left) 调用这个public
*/