题目描述:
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
算法思想:
用了树的基本概念,两个函数,每个函数分别有一次递归,HasSubtree这个函数查看根结点是否相同,如果相同执行fun函数,fun函数的作用是在判定好
根结点相同后,查看这个根结点下面是不是和Tree2一样。
注意:
在fun()函数之中要加这两句
// 如果Tree2已经遍历完了都能对应的上,返回true
if(pRoot2==NULL)
{
return true;
}
//如果Tree2还没有遍历完,Tree1却遍历完了。返回false
if(pRoot1==NULL)
{
return false;
}
代码(牛客网):
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
int i,j;
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
bool result=false;
if((pRoot1!=NULL)&&(pRoot2!=NULL))
{
if(pRoot1->val==pRoot2->val)
{
result=fun(pRoot1,pRoot2);
}
if(result==false)
{
result=HasSubtree(pRoot1->left,pRoot2);
}
if(result==false)
{
result=HasSubtree(pRoot1->right,pRoot2);
}
}
return result;
}
bool fun(TreeNode *pRoot1,TreeNode *pRoot2)
{
bool temp=false;
int i=2;
//注意一定要加这两句而且顺序不能变,判断是否遍历完成
//如果Tree2已经遍历完了都能对应的上,返回true
if(pRoot2==NULL)
{
return true;
}
//如果Tree2还没有遍历完,Tree1却遍历完了。返回false
if(pRoot1==NULL)
{
return false;
}
if(pRoot1->val==pRoot2->val)
{
if(fun(pRoot1->left,pRoot2->left))
{
if(fun(pRoot1->right,pRoot2->right))
{
temp=true;
}
}
}
else{
temp=false;
return temp;
}
return temp;
}
};