ZZULIOJ 1900: 985的“树”难题 （DFS+并查集）

1900: 985的“树”难题

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 32   Solved: 8

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

2
3 1
1 2
2 3
2 1
1 1

Sample Output

YES
2 1 0
1 1 2
NO

HINT

Source

hpu

大神说这是签到题，我就笑笑，不说话。

这道题利用DFS和并查集，并用了邻接表储存数据。

DFS用来搜索子节点，并查集用来记录子节点的个数

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
struct node
{
	int u,v,next;
}edge[40040];
int pre[10010],ran[10010],head[10010],vis[10010];
int num;
void add(int u,int v)//建立邻接表 
{
	edge[num].u=u;
	edge[num].v=v;
	edge[num].next=head[u];
	head[u]=num++;
}
int find(int x)
{
	if(x==pre[x])
	return x;
	else
	{
		ran[pre[x]]++;//记录树的深度 
		return find(pre[x]);
	}
}
void join(int x,int y)
{
	pre[y]=x;
}
void dfs(int u)
{
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		if(!vis[edge[i].v]&&edge[i].v!=pre[u])
		{
			join(u,edge[i].v);
			vis[edge[i].v]=1;
			dfs(edge[i].v);//搜索下一个子节点 
		}
	}
}
int main()
{
	int T,root,n,u,v,i;
	scanf("%d",&T);
	while(T--)
	{
		memset(head,-1,sizeof(head));
		memset(vis,0,sizeof(vis));
		num=0;
		scanf("%d%d",&n,&root);
		for(i=1;i<=n;i++)
		{
			pre[i]=i;
			ran[i]=0;
		}
		for(i=1;i<n;i++)
		{
			scanf("%d%d",&u,&v);
			add(u,v);
			add(v,u);
		}
		vis[root]=1;
		dfs(root);
		bool flag=true;
		if(flag)
		{
			for(i=1;i<=n;i++)//判断是不是树 
			{
				if(find(i)!=root)
				{
					flag=false;
					break;
				}
			}
		}
		if(flag)
		{
			printf("YES\n");
			for(i=1;i<=n;i++)
			{
				printf("%d",ran[i]);
				if(i==n)
			    printf("\n");
				else
			    printf(" ");
			}
			
			for(i=1;i<=n;i++)
			{
				printf("%d",pre[i]);
				if(i==n)
				printf("\n");
				else
				printf(" ");
			}
		}
		else
		{
			printf("NO\n");
		}
	}
	return 0;
}