求树的子结构

好长时间不更新博客了，今天看到一道以前做过的题（题目来自《剑指offer》），再练一下：

我的思路如下：

判断B是不是A的子结构，就是要判断B是不是和A中的一部分结构是不是完全相同，需要一个节点一个节点的判断，只要有一个节点不同，就返回false。这部分代码如下：

bool isEqual(TreeNode *head1,TreeNode *head2)
{
	if(head2 == NULL)
		return true;
	if(head1 == NULL)
		return false;
	if(head1->data == head2->data)
		return isEqual(head1->left,head2->left) && isEqual(head1->right,head2->right);
	return false;<pre name="code" class="cpp">

}

然后将A中的节点依次作为根节点同B的根节点进行比较：每一次比较都执行依次isEqual（），若isEqual（）返回true，则整体返回true；否则将A的左子树和右子树同B比较。代码如下：

bool isSubTree(TreeNode *head1,TreeNode *head2)
{
	if(head2 == NULL)
		return true;
	if(head1 == NULL)
		return false;
	if(isEqual(head1,head2))
		return true;
	return isSubTree(head1->left,head2) || isSubTree(head1->right,head2);
}