1、题目
输入两棵二叉树A和B,判断B是不是A的子结构。二叉树结点的定义如下:
struct BinaryTreeNode
{
int m_nValue;
BinaryTreeNode* m_pLeft;
BinaryTreeNode* m_pRight;
};
树的子结构形式如图1所示:
图1、两棵二叉树A和B,右边树B是左边树A的子结构
2、解题思路
分两步来:第一步,在树A中找到和B的根结点的值一样的结点R,第二步再判断树A中以R为根结点的子树是不是包含和树B一样的结构。
第一步在树A中查找与根结点的值一样的结点,这实际上就是树的遍历。对二叉树的遍历可以用递归和循环方法。
3、参考代码
//判断树的子结构
bool HasSubtree(BiaryTreeNode* pRoot1, BinaryTreeNode* pRoot2)
{
bool flag=false;
if(pRoot1 != NULL && pRoot2 != NULL)
{
if(pRoot1->m_nValue == pRoot2->m_nValue)
{
flag = DoesTree1HaveTree2(pRoot1,pRoot2);
}
if(!flag)
flag=HasSubtree(pRoot1->m_pLeft,pRoot2);
if(!flag)
flag=HasSubtree(pRoot1->m_pRight,pRoot2);
}
return flag;
}
bool DoesTree1HaveTree2(BiaryTreeNode* pRoot1, BinaryTreeNode* pRoot2)
{
if(pRoot1 == NULL)
return false;
if(pRoot2 == NULL)
return true;
if(pRoot1->m_nVlaue != pRoot2->m_nValue)
return false;
return DoesTree1HaveTree2(pRoot1->m_pLeft, pRoot2->m_pLeft) &&
DoesTree1HaveTree2(pRoot1->m_pRight, pRoot2->m_pRight);
}