《剑指offer》面试题18——树的子结构(C++)

题目描述:
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)

解题思路:
链接:https://www.nowcoder.com/questionTerminal/6e196c44c7004d15b1610b9afca8bd88
来源:牛客网

思考:
1.首先需要递归pRoot1树,找到与pRoot2根一样的节点,这需要一个遍历
2.找到相同的根节点后,要判断是否子树,仍需要一个一个遍历对比
树的遍历我们一般就用递归来做,那么根据分析需要两个递归函数如下:

代码:

bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
    bool result = false;

        if(pRoot2!=NULL&&pRoot1!=NULL)
     {
            if(pRoot1->val==pRoot2->val)   //头结点相同
            {
                result = DoesTree1haveTree2(pRoot1,pRoot2);    //比较子节点
            }
            if(!result)
            {
                result = HasSubtree(pRoot1->left, pRoot2);     //继续找和2头结点相同的结点
            }
            if(!result)
            {
                result=  HasSubtree(pRoot1->right, pRoot2);
            }
     }

     return result;
}

bool DoesTree1haveTree2(TreeNode* pRoot1, TreeNode* pRoot2)
{

    if(pRoot2==NULL)
        return true;
    if(pRoot1==NULL)
        return false;
    if(pRoot1->val!=pRoot2->val)
        return false;

    return DoesTree1haveTree2(pRoot1->left, pRoot2->left)&&DoesTree1haveTree2(pRoot1->right, pRoot2->right);
}

猜你喜欢

转载自blog.csdn.net/zhenaoxi1077/article/details/80412639