剑指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)
    {
		bool result=false;
        if(NULL!= pRoot1 && NULL != pRoot2)
        {
        	if(pRoot1->val == pRoot2->val)
        	{
                result = DoesTree1HaveTree2(pRoot1, pRoot2);
            }
            if(!result)
            {
                result = HasSubtree(pRoot1->left, pRoot2);
            }
            if(!result)
            {
        		result = HasSubtree(pRoot1->right, pRoot2);
        	}
        }
        return result;
    }
    bool DoesTree1HaveTree2(TreeNode* pRoot1, TreeNode* pRoot2)
    {
    	if(NULL == pRoot2)
    	{
            return true;
        }
        if(NULL == pRoot1)
        {
            return false;
        }
        if(pRoot1->val != pRoot2->val)
        {
            return false;
        }
        return DoesTree1HaveTree2(pRoot1->left, pRoot2->left) && DoesTree1HaveTree2(pRoot1->right, pRoot2->right);
    }
};

运行结果
运行时间：4ms
占用内存：496k

解题思路
第一层递归是找与 B 树根结点相同的结点。 HasSubtree 函数
第二层递归是找到与 B 树根结点相同的结点后，判断 B 左右子树是否属于与该节点下左右子树