输入两棵二叉树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;
        }
    }
};


你可能感兴趣的:(acm,题)