剑指offer| |树的子结构

题目:输入两个二叉树A和B,判断A是不是B的子结构。

树的子结构
树的子结构

如图下面这颗树就是上面的树的子结构。

思路:要查找树A中是否存在和树B结构一样的子树,我们可以分为两步:第一步在树A中找到和B的根节点的值一样的结点R,第二步在判断树A中以R为根节点的子树是不是包含和树B一样的结构。

第一步在树A中找与树B根节点的值一样的结点,这实际上就是树的遍历。
第二步是判断树A中以R为根节点的子树是不是和树B具有相同的结构。
递归来考虑:如果结点R的值和树B的根的值不一样,则以R为根节点的子树和树B肯定不具有相同的结点;如果他们的节点值相同,则递归判断他们各自的左右子树的结点的值是不是相同的。递归的终止条件是我们到达了树A或者树B的叶子节点。

bool DoTree1HaveTree2(BTNode* root1, BTNode* root2)
{
    if (root1 == NULL)
    {
        return false;
    }

    if (root2 == NULL)
    {
        return true;
    }

    if (root1->_data != root2->_data)
    {
        return false;
    }

    return DoTree1HaveTree2(root1->_left, root2->_left) && DoTree1HaveTree2(root1->_right, root2->_right);
}

bool HasSubTree(BTNode* root1, BTNode* root2)
{
    bool result = false;

    if (root2 == NULL)
    {
        return true;
    }

    if (root1 == NULL)
    {
        return false;
    }

    if (root1->_data == root2->_data)
    {
        result = DoTree1HaveTree2(root1, root2);
    }

    if (!result)
    {
        result = HasSubTree(root1->_left, root2);
    }

    if (!result)
    {
        result = HasSubTree(root1->_right, root2);
    }

    return result;
}

猜你喜欢

转载自blog.csdn.net/qq_40399012/article/details/82260986