树的子结构--《剑指offer》面试题18

题目

  输入两棵二叉树A和B，判断B是不是A的子结构。如图：A中有一部分子树的结构和B是一样的，因此B是A的子结构。

思路

  要查找树A中是否存在和树B结构一样的子树，分成两步：第一，在树A中找到和B的根结点的值一样的结点R；第二，判断树A中以R为根结点的子树是不是包含和树B一样的结构。

  判断树A中以R为根结点的子树是不是和树B具有相同的结构：
  用递归方法。如果结点R的值和树B的根结点不相同，则以R为根结点的子树和树B肯定不具有相同的结点；如果它们的值相同，则递归地判断它们各自的左右结点的值是不是相同。递归终止的条件是我们遍历完了树A或者树B的叶结点。

代码

#include <iostream>

struct BinaryTreeNode
{
    int  m_nValue;
    BinaryTreeNode*  m_pLeft;
    BinaryTreeNode*  m_pRight;
};

// 判断二叉树1是否包含二叉树2
bool DoseTree1HaveTree2(BinaryTreeNode* pRoot1, BinaryTreeNode* pRoot2)
{
    //  如果遍历完二叉树2的全部结点，返回true
    if (pRoot2 == nullptr)
        return true;

    //  如果遍历完二叉树1的全部结点，并且二叉树2未遍历结束，返回false
    if (pRoot1 == nullptr)
        return false;

    //  如果遍历过程中有二叉树的结点不相同，遍历结束，返回false
    if (pRoot1->m_nValue != pRoot2->m_nValue)
        return false;

    //  递归遍历二叉树的所有左右子树。
    return DoseTree1HaveTree2(pRoot1->m_pLeft, pRoot2->m_pLeft) &&
           DoseTree1HaveTree2(pRoot1->m_pRight, pRoot2->m_pRight);
}

//  判断两棵树的根结点是否相同
bool HasSubtree(BinaryTreeNode* pRoot1, BinaryTreeNode* pRoot2)
{
    bool result = false;

    // 两个根结点不能为空
    if (pRoot1 != nullptr && pRoot2 != nullptr)
    {
        //  如果找到结点与另一子树的根结点相同，则判断是否包含此子树
        if (pRoot1->m_nValue == pRoot2->m_nValue)
            result = DoseTree1HaveTree2(pRoot1,pRoot2);

        //  如果未找到，从左子树中查找
        if (!result)
            result = HasSubtree(pRoot1->m_pLeft, pRoot2);

        //  如果未找到，从右子树中查找
        if (!result)
            result = HasSubtree(pRoot1->m_pRight, pRoot2);
    }

    return result;
}

int main()
{

    return EXIT_SUCCESS;
}