剑指OFFER之树的子结构（九度OJ1520） 本地通过

剑指OFFER之树的子结构（九度OJ1520）

阅读目录

题目描述：

解题思路：

全部代码：

回到顶部

题目描述：

输入两颗二叉树A，B，判断B是不是A的子结构。

输入：

输入可能包含多个测试样例，输入以EOF结束。

对于每个测试案例，输入的第一行一个整数n,m(1<=n<=1000,1<=m<=1000)：n代表将要输入的二叉树A的节点个数（节点从1开始计数），m代表将要输入的二叉树B的节点个数（节点从1开始计数）。接下来一行有n个数，每个数代表A树中第i个元素的数值，接下来有n行，第一个数Ki代表第i个节点的子孩子个数，接下来有Ki个树，代表节点i子孩子节点标号。接下来m+1行，与树A描述相同。

输出：

对应每个测试案例，

若B是A的子树输出”YES”(不包含引号)。否则，输出“NO”（不包含引号）。

样例输入：

复制代码

复制代码

7 3

8 8 7 9 2 4 7

2 2 3

2 4 5

0

0

2 6 7

0

0

8 9 2

2 2 3

0

0

1 1

2

0

3

0

复制代码

复制代码

样例输出：

YES

NO

提示：

B为空树时不是任何树的子树。

#include <stdio.h>
#include <stdlib.h>
#include<assert.h>
#define MAXSIZE 1005
typedef struct treenode{
int val;
int left;
int right;
}TreeNode;
typedef struct treearr{
TreeNode arr[MAXSIZE];
}treearr;
int test(treearr *t1,treearr *t2,int n1,int n2)
{
if(n2==0)
return 1;
if(n1==0)
return 0;
if(t1->arr[n1].val!=t2->arr[n2].val)
return 0;
return test(t1,t2,t1->arr[n1].left,t2->arr[n2].left)&&
test(t1,t2,t1->arr[n1].right,t2->arr[n2].right);//,t1->t1->arr[n1].left->val,
//  t2->arr[n2].left->val)&&test(t1->arr[n1].right,t2->arr[n2].left);
}
int testequal(treearr *t1,treearr *t2,int n)
{
int result=0;
if(n!=0&&t2->arr[0].val==0)
{
if(t1->arr
.val==t2->arr[1].val)
{
result=test(t1,t2,n,1);
}
if(!result)
result=testequal(t1,t2,t1->arr
.left);
if(!result)
result=testequal(t1,t2,t1->arr
.right);
}
return result;
}

int main()
{
//assert(freopen("text.txt","r",stdin));
int n,m,nchild;
while(scanf("%d %d",&n,&m)!=EOF)
{
treearr *t1=(treearr*)malloc(sizeof(treearr));
treearr *t2=(treearr*)malloc(sizeof(treearr));
for(int i=0;i<1000;i++)
{
t1->arr[i].val=0;
t1->arr[i].left=0;
t1->arr[i].right=0;

t2->arr[i].val=0;
t2->arr[i].left=0;
t2->arr[i].right=0;
}
for(int i=1;i<=n;i++)
{
scanf("%d",&t1->arr[i].val);
}
for(int i=1;i<=n;i++)
{
scanf("%d",&nchild);
if(nchild==2)
{
scanf("%d",&t1->arr[i].left);
scanf("%d",&t1->arr[i].right);
}
else if(nchild==1)
{
scanf("%d",&t1->arr[i].left);
}
else
{
}
}
for(int i=1;i<=m;i++)
{
scanf("%d",&t2->arr[i].val);
}
for(int i=1;i<=m;i++)
{

scanf("%d",&nchild);
if(nchild==2)
{
scanf("%d",&t2->arr[i].left);
scanf("%d",&t2->arr[i].right);
}
else if(nchild==1)
{
scanf("%d",&t2->arr[i].left);
}
else
{
}
}
if(testequal(t1,t2,1))
printf("YES\n");
else
printf("NO\n");
}
return 0;
}