面试题18：树的子结构

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

struct BinaryTreeNode{
    int value;
    BinaryTreeNode* left;
    BinaryTreeNode* right;
    BinaryTreeNode(int data=int()):value(data),left(NULL),right(NULL)
    {}
};

分析

要判断 B 是不是 A 的子结构，也就是要判断 B 是不是 A 的子树。我们可以先遍历 A，如果某一个 S 节点的 value 和 B 树的根节点相同，就判断 S 和 B 的左右子树是否相等，如果相等，B 就是 A 的子结构；否则跳过 S 继续建立 A，直到遍历完成。

代码如下：

bool HasSubTree(BinaryTreeNode* root1,BinaryTreeNode* root2)
{
    if(root1==NULL)
    {
        if(root2==NULL)
            return true;
        else
            return false;
    }
    if(root2==NULL)
        return true;
    if(root1->value==root2->value)
    {
        if(HasSubTree(root1->left,root2->left) &&
                HasSubTree(root1->right,root2->right))
        {
            return true;
        }
        else
        {
            if(HasSubTree(root1->left,root2) || HasSubTree(root1->right,root2))
            {
                return true;
            }
            else
            {
                return false;
            }
        }
    }
    else
    {
        if(HasSubTree(root1->left,root2) || HasSubTree(root1->right,root2))
        {
            return true;
        }
        else
        {
            return false;
        }
    }
}

如果你有任何想法或是可以改进的地方，欢迎和我交流！

完整代码及测试用例在github上：点我前往