面试题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上：点我前往