题目
给定两个非空二叉树 s 和 t,检验 s 中是否包含和 t 具有相同结构和节点值的子树。s 的一个子树包括 s 的一个节点和这个节点的所有子孙。s 也可以看做它自身的一棵子树。
示例 1:
给定的树 s:
3
/ \
4 5
/ \
1 2
给定的树 t:
4
/ \
1 2
返回 true,因为 t 与 s 的一个子树拥有相同的结构和节点值。
示例 2:
给定的树 s:
3
/ \
4 5
/ \
1 2
/
0
给定的树 t:
4
/ \
1 2
返回 false。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subtree-of-another-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
这是一道简单题,但是没有直接AC,用例在90%左右的时候执行出错。
未看题解前
一开始看到题目,还以为是个简单的遍历二叉树的题。然后仔细往下看了看,发现并不是这么简单。
- 比较2棵树是否一致,可以采用前序遍历法对两棵二叉树进行遍历,每次递归都比较一下两棵树的
val
。 - 之前做过一道题,要求判断t是否属于s,即无论s下是否还存在子节点,t都属于s,即做以下判断:
if (s == nullptr && t == nullptr) {
return true;
}
- 但是这里并没这么简单,需要的是s的子节点的所有子孙节点 == t,所以在这需要加上一层判断:
if (s == nullptr && t == nullptr) {
return true;
}
// 新加判断
if (s == nullptr || t == nullptr) {
return false;
}
- 递归遍历该二叉树。
- 以上分析会遗漏一个场景,当t已经遍历到了中间点时,当前的s节点若与t的根节点一致,则无法判断当前s节点是否是符合条件的子树 代码如下:
写完以后发现自己想的有有些繁琐,应该不是正解,遂看了题解。
class Solution {
public:
bool result = false;
TreeNode *t1;
bool isSubtree(TreeNode* s, TreeNode* t) {
t1 = t;
return dis(s, t);
}
bool dis(TreeNode* s, TreeNode* t) {
if (result) {
return result;
}
if (s == nullptr && t == nullptr) {
return true;
}
if (s == nullptr || t == nullptr) {
return false;
}
if (s->val == t->val) {
bool ltmp = dis(s->left, t->left);
bool rtmp = dis(s->right, t->right);
if (t == t1 && ltmp && rtmp) {
result = true;
}
return result ? result : ltmp && rtmp;
} else {
bool ltmp = dis(s->left, t);
bool rtmp = dis(s->right, t);
if (t == t1 && ltmp && rtmp) {
result = true;
}
return result ? result : ltmp && rtmp;
}
}
};
看了题解后
二叉树问题不一定要想的那么复杂,如果想复杂了,建议重头开始整理思路。
看了题解后,发现自己想的太复杂了。对于二叉树的结构只需要明确一点:一个二叉树由其左子树和右子树构成。需要判断是否存在子树,只需要判断以下三点:
- t是否存在s的左子树内
- t是否存在s的右子树内
- t是否等于s
明确了以上3个条件后,就很容易写代码了。
代码
class Solution {
public:
bool isSubtree(TreeNode* s, TreeNode* t) {
// 加这一层判断防止下面递归访问非法内存
if (s == nullptr) {
return false;
}
return isSameTree(s, t) || isSubtree(s->left, t) || isSubtree(s->right, t);
}
bool isSameTree(TreeNode *s, TreeNode *t) {
if (s == nullptr && t == nullptr) {
return true;
}
if (s == nullptr || t == nullptr || s->val != t->val) {
return false;
}
return isSameTree(s->left, t->left) && isSameTree(s->right, t->right);
}
};