题目
给定两个二叉树,编写一个函数来检验它们是否相同。
如果两个树在结构上相同,并且节点具有相同的值,则认为它们是相同的。
我的想法
我现阶段就只是做简单题嘛,整体的逻辑先慢慢清晰起来。
题目即要求对两个二叉树遍历,每个节点都一样就返回ture。遍历的话,我脑子里有前序遍历,中序遍历,后序遍历,我觉得前序遍历我更熟点,就先写一个前序遍历的函数出来。
然后才想的是边界条件的设置(我觉得这不是好习惯)。就是对对应节点是否为空的讨论。if语句,搞。
用乘法实现结果的记录。
代码
/**
* 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:
int traverse(TreeNode* p, TreeNode* q){
if (p==nullptr && q==nullptr){
return 1;
}
if ((p==nullptr && q!=nullptr)||(p!=nullptr && q==nullptr)){
return 0;
}
int f;
if (p->val == q->val){
f = 1;
}
else{
f = 0;
}
return f * traverse(p->left, q->left) * traverse(p->right, q->right);
}
bool isSameTree(TreeNode* p, TreeNode* q) {
int flag = traverse(p, q);
if (flag){
return true;
}
else{
return false;
}
}
};
执行结果如上。倒是能够通过,就是内存消耗过多。
下面是题解的代码:
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);
}
}
};
反思
前序遍历二叉树第一次自己写出来了,还不熟练,要更加多加练习。
内存消耗过多可能是由于我创建新变量太多吧。官方题解在函数中甚至一个新变量都没有创建,而且逻辑上看起来很清晰。
题解中对方法的分类:“广度优先搜索”;“深度优先搜索”。这两种方法的实现形式有些不同,而我对这两种的实现还很陌生。