剑指offer第17题：树的子结构

题目描述

树的子结构

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

分析

有点小复杂，让手动判断一个试试，看看过程。

如图所示，左侧为树A，右侧为树B,要查找树A中是否存在和树B结构一样的子树，我们可以分成两步：第一步在树A中找到和B的根结点的值一样的结点C,第二步再判断树A中以C为根结点的子树是不是包含和树B一样的结构。如果不是，则用A的左子树代替A,若还是，则用A的右子树代替A

源码

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
//先在A中找到一个节点值与B的根节点的值相同的节点C，判断B是不是C的同根子结构（也就是B的根节点固定在c的根节点）
class Solution {
public:
    bool isSameRootSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if(pRoot1 == NULL && pRoot2 == NULL) return true;//说明遍历完了所有节点，都相同，返回true。
        if(pRoot1 == NULL) return false;//此时C中已经遍历完了，但是B中还节点，说明B不是C的子树。
        if(pRoot2 == NULL) return true;//此时B已经遍历完了，说明B的节点都符合C的结构，返回True.
        if(pRoot1->val == pRoot2->val)//若当前节点值相同，则比较其左右节点是否也相同。
        return isSameRootSubtree(pRoot1->left,pRoot2->left)&&isSameRootSubtree(pRoot1->right,pRoot2->right);
        return false;//一旦发现对应位置不相同，则说明B不是C的同根子结构。
    }
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if(!pRoot1 || !pRoot2) return false;//A为空，那么任何数都不会是其子结构，若B为空，返回空。
        if(pRoot1->val == pRoot2->val)//若根节点相同，则判断pRoot2是不是pRoot1的同根子结构，
        {
            if(isSameRootSubtree(pRoot1,pRoot2)) return true;//若是，则返回true。
        }
        return HasSubtree(pRoot1->left,pRoot2)||HasSubtree(pRoot1->right,pRoot2);//若不是则换节点左右子树接着找。
    }
};