树的子结构

题目描述：

输入两棵二叉树A、B，判断B是不是A的子结构。（ps：我们约定空树不是任意一个树的子结构）

 

解题思路：

遍历A树，找到与B树根节点相同的结点，再将它们的左右子树比较，直到B树为空。则返回true。

class Solution {
public:
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if(pRoot1 == NULL || pRoot2 == NULL)    //前序遍历的出口
            return false;
        //前序遍历主树pRoot1各结点，从根节点到左子树再到右子树
        if(is_same(pRoot1, pRoot2))
            return true;
        else    //判断当前结点的左子树是否为子树pRoot2,再判断右子树是否为子树pRoot2
            return HasSubtree(pRoot1->left, pRoot2) || HasSubtree(pRoot1->right, pRoot2);
    }
    //确定父结点后，开始同时扫描pRoot1和pRoot2的各结点是否相等
    bool is_same(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if(pRoot2 == NULL)    //如果遍历到pRoot2的空结点，返回true
            return true;
        if(pRoot1 == NULL)    //如果pRoot1为空，而pRoot2不为空，说明不同，返回false
            return false;
        if(pRoot1->val == pRoot2->val)
            return is_same(pRoot1->left, pRoot2->left) && is_same(pRoot1->right, pRoot2->right);
        else
            return false;    //不相同返回false
    }
};