leetcode 572. 另一个树的子树 easy
题目描述:
给定两个非空二叉树 s 和 t,检验 s 中是否包含和 t 具有相同结构和节点值的子树。s 的一个子树包括 s 的一个节点和这个节点的所有子孙。s 也可以看做它自身的一棵子树。
示例 2:
给定的树 s:
3
/ \
4 5
/ \
1 2
/
0
给定的树 t:
4
/ \
1 2
返回 false。
解题思路:
注意子树和剑指offer的子结构的不同。(上述示例可以看出)
深度优先搜索枚举 s中的每一个节点,判断这个点的子树是否和 t相等。如何判断一个节点的子树是否和 t相等呢,我们又需要做一次深度优先搜索来检查,即让两个指针一开始先指向该节点和 t的根,然后「同步移动」两根指针来「同步遍历」这两棵树,判断对应位置是否相等。
代码:
class Solution {
public:
bool isSubtree(TreeNode* s, TreeNode* t) {
if(s==nullptr)
return false;
if(issame(s,t)) return true;
return isSubtree(s->left,t) || isSubtree(s->right,t);
}
bool issame(TreeNode *s, TreeNode *t){
if(s==nullptr && t==nullptr) // 递归出口 如果最后均遍历到空结点,返回true
return true;
if(s==nullptr || t==nullptr) // 只有一个为空,则为false
return false;
if(s->val != t->val)
return false;
return issame(s->left,t->left) && issame(s->right,t->right);
}
};
剑指offer. 26 树的子结构
题目描述:
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
解题思路:
注意跟上面子树的差别,leetcode的示例2也是子结构
// 遍历树A, 对于遍历到的每一个节点,询问以当前节点为根的子树,是不是和树B有相同的结构
代码:
class Solution {
public:
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
if(!pRoot1 || !pRoot2)
return false;
if(issame(pRoot1,pRoot2))
return true;
return HasSubtree(pRoot1->left,pRoot2) || HasSubtree(pRoot1->right,pRoot2);
}
bool issame(TreeNode * proot1, TreeNode * proot2){
if(proot2==nullptr) // proot2遍历完成(关键语句)
return true;
if(proot1==nullptr) //proot1遍历完了,但是proot2没遍历完
return false;
if(proot1->val != proot2->val)
return false;
return issame(proot1->left,proot2->left) && issame(proot1->right,proot2->right);
}
};