判断一颗二叉树是否是另一颗树的子树

• 题目：输入两棵二叉树A和B判断B是不是A的子树，二叉树节点定义如下

struct BinaryTree
{
    BinaryTree(char data)
    :_pLeft(NULL)
    , _pRight(NULL)
    , _data(data)
    {}
    BinaryTree *_pLeft;
    BinaryTree *_pRight;
    char _data;
};

附上创建树的代码，方便测试

void CreateBinaryTree(BinaryTree *&pRoot, char *str,size_t size, size_t &index)
{
    if (index < size && str[index] != '#')
    {
        pRoot = new BinaryTree(str[index]);
        CreateBinaryTree(pRoot->_pLeft, str, size, ++index);
        CreateBinaryTree(pRoot->_pRight, str, size, ++index);
    }
}

思路：可以分两步，第一步在在A中找和B的跟节点的值一样的结点R，第二步在判断A中以R为根节点的子树是不是包含和树B一样的结构。

• 代码实现

bool IsSameSubTree(BinaryTree *pAroot, BinaryTree *pBroot)//判断相同根节点的子结构是否相同
{
    if (NULL == pBroot)//B树为空了还没有返回证明A中有与B相等的子结构
        return true;
    if (NULL == pAroot)//A树为空了证明没有与B相同的子结构
        return false;
    if (pBroot->_data != pAroot->_data)//有节点不相等证明不是子结构
        return false;
    //相等继续判断其他节点
    return (IsSameSubTree(pAroot->_pLeft, pBroot->_pLeft) && \
        IsSameSubTree(pAroot->_pRight, pBroot->_pRight));
}
//判断B是否是A的子树
bool HassubTree(BinaryTree *pAroot,BinaryTree *pBroot)//寻找与B的根节点相同的结点
{
    bool result = false;
    if (NULL == pBroot || NULL == pAroot)
        return result;
    if (pBroot->_data == pAroot->_data)
        result = IsSameSubTree(pAroot, pBroot);//判断相同结点的子结构是否相等
    if (!result)//子结构不相等再继续寻找与B根节点相同的节点
        result = HassubTree(pAroot->_pLeft, pBroot);
    if (!result)
         result = HassubTree(pAroot->_pRight, pBroot);
    return result;
}

• 测试用例

void subtreetest()
{
    BinaryTree *pAroot;
    BinaryTree *pBroot;
    BinaryTree *pCroot;
    char *astr = "aab##cd##e##e";
    char *bstr = "ab##c";
    char *cstr = "cd##f";
    size_t index = 0;
    CreateBinaryTree(pBroot, bstr, strlen(bstr), index);
    index = 0;
    CreateBinaryTree(pAroot, astr, strlen(astr), index);
    bool ret = HassubTree(pAroot, pBroot);
    cout << ret << endl;
    index = 0;
    CreateBinaryTree(pCroot, cstr, strlen(cstr), index);
    ret = HassubTree(pAroot, pCroot);
    cout << ret << endl;
}