《剑指offer》树的子结构

一、题目描述

输入两颗二叉树A，B，判断B是不是A的子结构。

二、输入描述

输入两棵树A,B。

三、输出描述

判断B是不是A的子结构，是返回true，否则返回false。

四、牛客网提供的框架

/*
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)
    {

    }
};

五、解题思路

使用递归的方法。判断树B的根节点值是否跟树A的相等。如果不相等，判断树B是否是树A的左子树的子结构或者是A的右子树的子结构。继续这样递归。

判断第一个节点相等时或者B节点为空，继续递归判断。当出现某个对于结点B的对应节点为不为NULL，而A对应的为NULL，折该子树不成立，继续判断下一次递归知道所有的结点都判断。

六、代码

/*
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)
    {
        if (pRoot1 == NULL || pRoot2 == NULL) return false;
        return treeAHasSubtreeTreeB(pRoot1, pRoot2) || HasSubtree(pRoot1->left, pRoot2) || HasSubtree(pRoot1->right, pRoot2);
    }

    bool treeAHasSubtreeTreeB(TreeNode* root1, TreeNode* root2)
    {

        if(!root1 && root2) return false;
        if(root1  && root2)
        {
            if(root1->val != root2->val) return false;
            bool leftSubtree = treeAHasSubtreeTreeB(root1->left, root2->left);
            bool rightSubtree = treeAHasSubtreeTreeB(root1->right, root2->right);
            return leftSubtree && rightSubtree;
        }
        return true;
    }
};

七、总结

开始做这道题时，由于误解了题目的意思，以为是要求B的根节点要对应A的根节点判断。结果第一个案例都没通过才发现。后来发现，上面只是其中的一次判断过程，不断地递归判断就可以实现。

相对Sicily里的题目，这些简单多了，只是基础中的基础。