输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
class Solution {
public:
bool issubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
if(pRoot1==NULL && pRoot2==NULL ) return true;
else if(pRoot1!=NULL && pRoot2==NULL) return true; //这里需要注意 如果判断a是否为b的子树,如果b==NULL a!=NULL 这种情况也返回true,这里和判断两个树是否相等还是有区别的;
else if(pRoot1==NULL &&pRoot2!=NULL) return false;
else if( pRoot1->val == pRoot2->val)
{
bool flag1= issubtree(pRoot1->left,pRoot2->left);
bool flag2= issubtree(pRoot1->right,pRoot2->right);
return flag1&&flag2;
}
else return false;
}
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
if(pRoot2==NULL || pRoot1==NULL) return false;
bool flag1,flag2;
if( issubtree(pRoot1,pRoot2) )
{
return true;
}
else
{
flag1=HasSubtree(pRoot1->left ,pRoot2);
flag2=HasSubtree(pRoot1->right ,pRoot2);
return flag1 || flag2;
}
}
};