剑指offer---树的子结构

题目：树的子结构

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

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/
class Solution {
public:
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {

    }
};

解题代码：

class Solution {
public:
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2) {
        bool res = false;

        if(pRoot1 != nullptr && pRoot2 != nullptr){
             //在Tree1中找到对应Tree2的根节点的节点
            if(pRoot1->val == pRoot2->val)
                //以这个根节点为为起点判断是否包含Tree2
                res = doesTree1HaveTree2(pRoot1, pRoot2);
            //未不到，以root的左孩子当作起点，判断是否包含Tree2
            if(!res)
                res = HasSubtree(pRoot1->left, pRoot2);
            //未不到，以root的右孩子当作起点，判断是否包含Tree2
            if(!res)
                res = HasSubtree(pRoot1->right, pRoot2);
        }
        return res;
    }

private:
    bool doesTree1HaveTree2(TreeNode* pRoot1, TreeNode* pRoot2){
        //如果Tree2已经遍历完了都能对应的上，返回true
        if(pRoot2 == nullptr)
            return true;
        //如果Tree2还没有遍历完，Tree1遍历完，返回false
        if(pRoot1 == nullptr)
            return false;
        //值不相等，返回false
        if(pRoot1->val != pRoot2->val)
            return false;
        //如果根节点对应的上，那么就分别去子节点里面匹配
        return doesTree1HaveTree2(pRoot1->left, pRoot2->left) && doesTree1HaveTree2(pRoot1->right, pRoot2->right);
    }
};