题目描述
输入两棵二叉树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 IsSubtree(TreeNode* p, TreeNode* q) //判断以该节点为根的子树是否是原树的子结构
{
if(q == NULL)
return true; //子树为空,递归结束
if(p == NULL) //p非空,q为空。不是子树
return false;
return p->val == q->val && IsSubtree(p->left, q->left) && IsSubtree(p->right, q->right); //递归判断,左子树与右子树
}
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
if(pRoot1 == NULL || pRoot2 == NULL)
return false;
return IsSubtree(pRoot1, pRoot2) || HasSubtree(pRoot1->left, pRoot2) || HasSubtree(pRoot1->right, pRoot2); //二叉树2或者是以树1根节点为子结构,或者是以其左子树为子结构,或者以其右子树为子结构
}
};