树的子结构
题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
子树与子结构的概念
子树:指包含树的某个结点下的所有结点的树结构。一个n个结点的二叉树有n棵子树。[CADB,A,DB,B,H,EHGM,GM…]等都是该二叉树的子树。
子结构:指包含树的某个结点,可包含左子树或者右子树,或者都左右子树包含的树结构。这样只要与树某一个结点有父子关系的树结构都是子结构。[C, DB, FCE, EGM…]等都是该二叉树的子结构。
递归
判断b树是不是a树的子结构?
我们先写出递归判断函数
- 如果b树为NULL,则遍历检查到了b树的左子树的尽头或者右子树的尽头,在左路或者右路的判断是正确的
- 如果a树为NULL,b树不为NULL,则判断错误,父树没有相应结点
- 如果a树根结点与b树根结点相同,接着判断a的左子树与b的左子树 && a的右子树与b的右子树
- 如果a树根结点与b树根结点不相同,判断a的左子树与b || a的右子树与b
在HasSubtree函数中我们直接return judge(pRoot1, pRoot2);
会出错,因为题目中的二叉树并不是排序二叉树,不能保证所有结点值不同,所以当结点判断失败后不能直接返回false,而要接着判断左子树和右子树是否存在满足的结构。
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
bool judge(TreeNode* a, TreeNode* b)
{
if (b == NULL)
{
return true;
}
if (a == NULL)
{
return false;
}
if (a->val == b->val)
{
return judge(a->right, b->right) && judge(a->left, b->left);
}
else
{
return judge(a->left, b) || judge(a->right, b);
}
}
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
if (pRoot2 == NULL || pRoot1 == NULL)
{
return false;
}
bool flag = judge(pRoot1, pRoot2);
if (!flag)
{
flag = HasSubtree(pRoot1->left, pRoot2) || HasSubtree(pRoot1->right, pRoot2);
}
return flag;
}
};