牛客网-剑指office-树的子结构

题目：输入两棵二叉树A，B，判断B是不是A的子结构。（ps：我们约定空树不是任意一个树的子结构）
思路：分两步。第一步找到树A中和树B根节点相同的节点R。
第二步用helper函数判断树A中以R为根节点的子树是否包含树B。

/*
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 (pRoot1 != nullptr&&pRoot2!=nullptr)
		{
			if (pRoot1->val == pRoot2->val)
				result = helper(pRoot1, pRoot2);
			if (result == false)
				result= HasSubtree(pRoot1->left, pRoot2);
			if (result == false)
				result = HasSubtree(pRoot1->right, pRoot2);
		}
		return result;
	}
	bool helper(TreeNode* t1, TreeNode* t2)
	{
		if (t2 == nullptr) return true;
		if (t1 == nullptr) return false;
		if (t1->val == t2->val)
			return helper(t1->left, t2->left) && helper(t1->right, t2->right);
		else return false;
	}
};