zzulioj-1900: 985的“树”难题

1900: 985的“树”难题

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 42   Solved: 13

Submit Status Web Board

Description

985给你一棵“树”以及它的根节点，要求你先判定它是否是一棵树，其次他想知道每个节点的“太子”数目以及它的父亲（root的话输出自己）。

“太子判定条件”：

一、若x是y的孩子节点，那么x是y的“太子”；

二、若x是y的“太子”且y是z的“太子”，那么x是z的“太子”。

Input

第一行输入一个整数t，代表有t组测试数据。

每组数据第一行输入两个整数n，root分别代表树的节点数目以及根节点的编号。

接下来n-1行，每行输出两个整数u，v代表u节点和v节点之间有一条树边。

注：1 <= t <= 20，1 <= n <= 1e4，1 <= root <= n，1 <= u，v <= n。

Output

对每一组数据，若给定的“树”不合法输出NO即可，反之输出YES，接下来输出占两行：

第一行输出n个整数代表每个节点的“太子”数目，

第二行输出n个整数代表每个节点的父亲节点编号。

输出顺序从1到n，每两个数之间有一个空格，最后一个数后面没有空格。

Sample Input

23 11 22 32 11 1

Sample Output

YES2 1 01 1 2NO

HINT

Source

hpu

题意很清晰 主要的问题没办法判断那个点是父节点，哪个点是子节点，这时候单单使用并查集做做不出来
需要借助邻接表来确定父子关系，这个难点解决就比较容易做了 

#include<cstdio>
#include<cstring>
#define M 100000+11
int pre[M],num[M],vis[M];
int n;
int head[M*3],c;
struct node
{
	int from;
	int to;
	int next;
}arr[M*3];
void add(int u,int v)
{
	arr[c].from=u;
	arr[c].to=v;
	arr[c].next=head[u];
	head[u]=c++;
}
void count()
{
	int i;
	for(i=1;i<=n;++i)
		pre[i]=i,num[i]=0;
}
void dfs(int x)
{
	int i;
	for(i=head[x];i!=-1;i=arr[i].next)
	{
		int o=arr[i].to;
		if(!vis[o])
		{
			pre[o]=x;//建立父子关系
			vis[o]=1;
			dfs(o); 
		} 
	}
}
void find(int i)
{
	while(pre[i]!=i)
	{
		i=pre[i];
		num[i]++;//存放每一个节点有多少个太子 
	}
}

int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		memset(head,-1,sizeof(head));//每个节点的地址都为-1 
		memset(vis,0,sizeof(vis));
		int root;
		scanf("%d%d",&n,&root);
		count();
		int a,b;
		c=0;//下标
		int g=n-1;
		while(g--)
		{
			scanf("%d%d",&a,&b);
			add(a,b);
			add(b,a);//建立邻接表 
		}
		vis[root]=1;
		dfs(root);//先跑一下dfs记录每一个节点的父节点
		int i,sum=0;
		for(i=1;i<=n;++i)
		{
			if(pre[i]==i)
			{
				++sum;
				if(sum>1)
					break;
			} 
		}
		if(sum==1)
		{
			printf("YES\n"); 
			for(i=1;i<=n;++i)
				find(i);
			for(i=1;i<n;++i)
				printf("%d ",num[i]);
			printf("%d\n",num[i]);
			for(i=1;i<n;++i)
				printf("%d ",pre[i]);
			printf("%d\n",pre[i]);	
		}
		else
			printf("NO\n");
	}
	return 0;
}