题目描述:
输入两棵二叉树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);
}