剑指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[n].val==t2->arr[1].val)
        {
            result=test(t1,t2,n,1);
        }
        if(!result)
        result=testequal(t1,t2,t1->arr[n].left);
        if(!result)
        result=testequal(t1,t2,t1->arr[n].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;
}