题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
解题思路
首先编写一个判定两棵树是否相同的函数isSame(TreeNode* p1, TreeNode*p2);
在HasSubTree函数中判断A的所有子树是否和B相同,若有相同的,则返回True。
代码
/*
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)
{
if(pRoot2==nullptr||pRoot1==nullptr)
return false;
return isSame(pRoot1,pRoot2)||HasSubtree(pRoot1->left,pRoot2)||HasSubtree(pRoot1->right,pRoot2);
}
bool isSame(TreeNode* p1, TreeNode*p2)
{
if(p1==nullptr&&p2)
return false;
if(p2==nullptr)
return true;
if(p1->val!=p2->val)
return false;
else
return isSame(p1->left,p2->left)&&isSame(p1->right,p2->right);
}
};