【hpu oj 1020 985的“树”难题 [DFS]】

点击打开链接

#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
#define M 10010
int n,root,f[M],fa[M],num[M];
vector<int> edge[M];
void init()
{
	for(int i=1;i<=n;i++)
	{
		f[i]=i;
		num[i]=0;
		fa[i]=-1;
		edge[i].clear();
	}
}
void dfs(int root,int father)
{
	if(edge[root].size()==1&&edge[root][0]==father)//只有一条 ，而且与父亲结点相连 ，那么它就是最下面的叶子 
	{
		num[root]=0;//其子节点数为0 
		return ;//递归结束 
	}
	for(int i=0;i<edge[root].size();i++)//否则找它的子节点 
	{
		int node=edge[root][i];
		if(node!=father)//只有 node!=father时 ，才是它的子节点 ，不然是父亲 
		{
			dfs(node,root);//以子节点node为根 ，root为父节点继续dfs 
			num[root]+=num[node]+1;
			fa[node]=root;//记录父节点 
		}
	}
}
int find(int x)
{
	if(x!=f[x])
		f[x]=find(f[x]);
	return f[x];
}
bool join(int x,int y)
{
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy)
	{
		f[fx]=fy;
		return true;
	}
	return false;
}
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		int ant=0,a,b,flag=1;
		scanf("%d%d",&n,&root);
		init();
		for(int i=1;i<n;i++)
		{
			scanf("%d%d",&a,&b);
			edge[a].push_back(b);
			edge[b].push_back(a);
			if(!join(a,b))
				flag=0;//判断是否有回路 
		}
		if(!flag)
		{
			printf("NO\n");
			continue;
		}
		for(int i=1;i<=n;i++)
		{
			if(f[i]==i)
				ant++;
			if(ant>1)
				break;//是否是一棵树 
		}
		if(ant>1)
		 	printf("NO\n");
		else
		{
			printf("YES\n");
			dfs(root,root);
			fa[root]=root;//根结点父亲是本身 
			for(int i=1;i<n;i++)
				printf("%d ",num[i]);
			printf("%d\n",num[n]);
			for(int i=1;i<n;i++)
				printf("%d ",fa[i]);
			printf("%d\n",fa[n]);
		}
	}
	return 0;
}