力扣 572 另一个树的子树
给定两个非空二叉树 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。
思路:
递归每个节点,判断其子树是否与树 t 结构相同。
代码:
class Solution
{
public:
bool isSubtree(TreeNode *s, TreeNode *t)
{
if (t == NULL)
return true;
if (s == NULL)
return false;
return isSub(s, t) || isSubtree(s->right, t) || isSubtree(s->left, t);
}
bool isSub(TreeNode *s, TreeNode *t)
{
if (s == NULL && t == NULL)
return true;
if (s == NULL || t == NULL)
return false;
if (s->val != t->val)
return false;
else
return isSub(s->right, t->right) && isSub(s->left, t->left);
}
};