题目描述
给定两个非空二叉树 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。
解题思路
自己的思路:先在S中找到树根T,然后判断子结构是否相同
vector<TreeNode*> fa;
void preOrder(TreeNode* s,TreeNode* t){
if(s!=NULL){
if(s->val == t->val) fa.push_back(s);
preOrder(s->left,t);
preOrder(s->right,t);
}
}
bool judge(TreeNode* root,TreeNode* t){
if(root == NULL && t == NULL) return true;
if((root == NULL && t != NULL) || (root != NULL && t == NULL)) return false;
if(root->val != t->val) return false;
bool j = judge(root->left,t->left);
if(j == false) return false;
else return judge(root->right,t->right);
}
bool isSubtree(TreeNode* s, TreeNode* t) {
if(t == NULL) return true;
preOrder(s,t);
for(int i=0;i<fa.size();i++){
if(judge(fa[i],t)) return true;
}
return false;
}
大佬的操作:
bool aux(TreeNode* s, TreeNode* t){
if (!s && !t) return true;
if (s && t){
if (s->val != t->val) return false;
return aux(s->left, t->left)&&aux(s->right, t->right);
}
return false;
}
bool isSubtree(TreeNode* s, TreeNode* t) {
if (s != nullptr){
if (aux(s, t)) return true;
if (isSubtree(s->left, t)) return true;
if (isSubtree(s->right, t)) return true;
}
return false;
}