题目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 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为空树时不是任何树的子树。

推荐指数：※

来源：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 fit
	if(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");
		else
			printf("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");
		else
			printf("NO\n");
    }
    return 0;
}