题目:
输入两颗二叉树A和B,判断B是不是A的子结构。
二叉树结点的定义如下:
struct BinaryTreeNode
{
int m_nValue;
BinaryTreeNode* m_pLeft;
BinaryTreeNode* m_right;
};
要查找树A是否存在和树B结构一样的子树,可以分成两步:
1、在树A中找到和B根结点的值一样的结点R
2、判断树A中以R为根结点的子树是否包含和树B一样的结构
bool DoesTree1HaveTree2(BinaryTreeNode* pRoot1,BianryTreeNode* pRoot2)
{
if(pRoot2==NULL)
return true;
if(pRoot1==NULL)
return false;
if(pRoot1->m_nValue!=pRoot2->m_nValue)
return false;
return DoesTree1HaveTree2(pRoot1->m_pLeft,pRoot2->m_pLeft) && DoesTree1HaveTree2(pRoot1->m_pRight,pRoot2->m_pRight);
}
bool HasSubtree(BinaryTreeNode* pRoot1,BinaryTreeNode* pRoot2)
{
bool result=false;
if(pRoot1!=NULL && pRoot2!=NULL)
{
//找到树A某结点的值与树B根结点相同,进行第二步判断
if(pRoot1->m_nValue==pRoot2->m_nValue)
result=DoesTree1HaveTree2(pRoot1,pRoot2);
if(!result)
result=HasSubtree(pRoot1->m_pLeft,pRoot2);
if(!result)
result=HasSubtree(pRoot1->m_right,pRoot2);
}
return result;
}
二叉树相关代码有大量指针操作,每一次使用指针的时候,我们都要检查指针是否为NULL,并对NULL的情况进行处理。