【剑指Offer】树的子结构

题目链接

题目描述

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

思路：

例如图中的两棵二叉树，由于A中有一部分子树的结构和B是一样的，因此B是A的子结构。

要查找树A中是否存在和树B结构一样的子树，可以分成两步：

1. 第一步在树A中找到和B的根节点的值一样的结点R；
2. 第二步再判断树A中以R为根结点的子树是不是包含和树B一样的结构。

第一步在树A中查找与根结点的值一样的结点，这实际上就是树的遍历。递归调用HasSubTree遍历二叉树A。如果发现某一结点的值和树B的头结点的值相同，则调用isSubtree，做第二步判断。

第二步是判断树A中以R为根结点的子树是不是和树B具有相同的结构。

以上参考:http://www.cnblogs.com/heyonggang/archive/2013/11/03/3405482.html

代码：

//struct TreeNode {
//    int val;
//    struct TreeNode *left;
//    struct TreeNode *right;
//    TreeNode(int x) :
//        val(x), left(NULL), right(NULL) {
//    }
//};
class Solution {
    bool isSubtree(TreeNode* pRootA, TreeNode* pRootB) {
        if (pRootB == NULL) return true;
        if (pRootA == NULL) return false;
        if (pRootB->val == pRootA->val) {
            return isSubtree(pRootA->left, pRootB->left) && isSubtree(pRootA->right, pRootB->right);
        } else {
            return false;
        }
    }
public:
    bool HasSubtree(TreeNode* pRootA, TreeNode* pRootB) {
        if (pRootA == NULL || pRootB == NULL) {
            return false;
        }
        return isSubtree(pRootA, pRootB) || HasSubtree(pRootA->left, pRootB) || HasSubtree(pRootA->right, pRootB);
    }
};