572. 另一个树的子树
写在最前面,其实5月每日一题都做了,但是有时候没时间写博客就没记录了,唉最近越来越忙。
上题目上题目
给定两个非空二叉树 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。
看完题目无非就是二叉树的遍历嘛,先找相同根节点,再遍历两个树是否一样
于是就有了下面的菜鸡代码(错误代码)
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
void DFS(TreeNode* s,int _val,TreeNode* &res)
{
if(s==NULL) return;
if(s->val==_val) res=s;
DFS(s->left,_val,res);
DFS(s->right,_val,res);
return;
}
void comp(TreeNode* s, TreeNode* t,bool &flag)
{
if(s==NULL&&t==NULL) return;
if(s==NULL) {flag=false;return;}
if(t==NULL) {flag=false;return;}
if(s->val!=t->val) {flag=false;return;}
comp(s->left,t->left,flag);
comp(s->right,t->right,flag);
}
bool isSubtree(TreeNode* s, TreeNode* t) {
int val=t->val;
TreeNode* res=NULL;
DFS(s,val,res);
if(res==NULL) return false;
bool flag=true;
comp(res,t,flag);
return flag;
}
};
跑跑前台没问题就交了
第162组测试数据错了
我去,我只考虑了找到相同的一个就行,忘了根节点重复相同的可能了。
解决办法:把所有相同根节点都存下了 vector 就行,再逐个比较
162测试数据
[1,null,1,null,1,null,1,null,1,null,1,null,1,null,1,null,1,null,1,null,1,2]
[1,null,1,null,1,null,1,null,1,null,1,2]
AC的代码(击败20%,我哭了)
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
void DFS(TreeNode* s,int _val,vector<TreeNode*>&res)
{
if(s==NULL) return;
if(s->val==_val) res.push_back(s);
DFS(s->left,_val,res);
DFS(s->right,_val,res);
return;
}
void comp(TreeNode* s, TreeNode* t,bool &flag)
{
if(s==NULL&&t==NULL) return;
if(s==NULL) {flag=false;return;}
if(t==NULL) {flag=false;return;}
if(s->val!=t->val) {flag=false;return;}
comp(s->left,t->left,flag);
comp(s->right,t->right,flag);
}
bool isSubtree(TreeNode* s, TreeNode* t) {
int val=t->val;
vector<TreeNode*>res;
DFS(s,val,res);
if(res.size()==0) return false;
bool sol=false;
for(int i=0;i<res.size();i++)
{
bool flag=true;
comp(res[i],t,flag);
if(flag==true)sol=true;
}
return sol;
}
};
好的解决思路很多,什么转字符串再KMP比较啊之类的。
这里放一个同思路的简洁代码(不愧是大佬,5行就vans了)
/**
* 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 isSubtree(TreeNode* s, TreeNode* t) {
if (!s) return false;
return isSubtree(s->left, t) || SameTree(s, t) || isSubtree(s->right, t);
}
bool SameTree(TreeNode* s, TreeNode* t) {
if (s == nullptr && t == nullptr) return true;
if (s == nullptr || t == nullptr) return false;
return s->val == t->val && SameTree(s->left, t->left) && SameTree(s->right, t->right);
}
};
5.7号结束收工 王者荣耀走起 哈哈哈哈 (写在这应该没人知道…