题目1520：树的子结构-九度

题目描述：

输入两颗二叉树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 38 8 7 9 2 4 72 2 32 4 5002 6 7008 9 22 2 3001 12030

样例输出：

YESNO

提示：

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

推荐指数：※

来源：http://ac.jobdu.com/problem.php?pid=1520

这道题本身算法木有难度，要注意几点：

1.vector使用要小心，超时。。

2.题目默认子树情况只有：无，左右，左的情况。不要想复杂了。

#include<iostream>#include<string.h>#include <stdio.h>#include<stdlib.h>using namespace std;const int N=1001;int a_edge[N][2],b_edge[N][2];int a_val[N],b_val[N];int n,m;bool dfs_sub(int a,int b){//sub tree is fitif(b==0)        return true;if(a_val[a]!=b_val[b]||false==dfs_sub(a_edge[a][0],b_edge[b][0])||false==dfs_sub(a_edge[a][1],b_edge[b][1]))return false;return true;}bool is_sub(){    int i;for(i=1;i<=n;i++){if(a_val[i]==b_val[1]){if(true==dfs_sub(i,1))return true;}}return false;}int main(){    while(scanf("%d%d",&n,&m)!=EOF){memset(a_val,0,sizeof(a_val));memset(b_val,0,sizeof(b_val));        int i,j,num;for(i=1;i<=n;i++)scanf("%d",&a_val[i]);for(i=1;i<=n;i++){scanf("%d",&num);for(j=0;j<num;j++)scanf("%d",&a_edge[i][j]);}for(i=1;i<=m;i++)scanf("%d",&b_val[i]);for(i=1;i<=m;i++){int num,tmp;scanf("%d",&num);for(j=0;j<num;j++)scanf("%d",&b_edge[i][j]);}if(m!=0&&n!=0&&true==is_sub())printf("YES\n");elseprintf("NO\n");    }    return 0;}

在附一个vector最后一个case超时的版本：

#include<iostream>#include<vector>#include<string>#include <stdio.h>#include<stdlib.h>using namespace std;vector<vector<int> > a_edge;vector<vector<int> > b_edge;int *a_val,*b_val;int n,m;bool dfs_sub(int a_node,int b_node){//sub tree is fit    if(a_val[a_node]==b_val[b_node]){if(b_edge[b_node].size()<a_edge[a_node].size()){int i,j;if(b_edge[b_node].size()==0)                return true;else{                if(dfs_sub(a_edge[a_node][0],b_edge[b_node][0])==true)return true;            }        }else if(b_edge[b_node].size()==a_edge[a_node].size()){int i;bool flag=false;for(i=0;i<b_edge[b_node].size();i++){if(false==dfs_sub(a_edge[a_node][i],b_edge[b_node][i]))return false;}return true;}    }    return false;}bool is_sub(){    int i;for(i=1;i<=n;i++){if(a_val[i]==b_val[1]){if(true==dfs_sub(i,1))return true;}}return false;}int main(){    while(scanf("%d%d",&n,&m)!=EOF){        a_val=new int [n+1];        b_val=new int [m+1];        a_edge.resize(n+1);        b_edge.resize(m+1);        int i,j;for(i=1;i<=n;i++)scanf("%d",&a_val[i]);for(i=1;i<=n;i++){int num,tmp;scanf("%d",&num);for(j=1;j<=num;j++){scanf("%d",&tmp);a_edge[i].push_back(tmp);}}for(i=1;i<=m;i++)scanf("%d",&b_val[i]);for(i=1;i<=m;i++){int num,tmp;scanf("%d",&num);for(j=1;j<=num;j++){scanf("%d",&tmp);b_edge[i].push_back(tmp);}}if(m!=0&&n!=0&&true==is_sub())printf("YES\n");elseprintf("NO\n");    }    return 0;}