【zzuli-oj】-1900- 985的“树”难题（DFS & 并查集&STL）

1900: 985的“树”难题

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 37   Solved: 11

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

题解：DFS用的好巧啊，不会啊！主要还是DFS理解不够透彻吧。

用vector 来存，也是很巧的。

先用并查集判断是不是一棵树，再跑一边DFS找到所有节点下面包括的节点数和其父节点。

#include<stdio.h>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
#define M 10000
int f[M+10];
int fa[M+10];
int num[M+10];
vector<int> edge[M+10];
int n,root;
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 u;
	scanf("%d",&u);
	while(u--)
	{
		int flag=1;
		scanf("%d %d",&n,&root);
		init();
		for(int i=1;i<n;i++)
		{
			int a,b;
			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;
		}
		int ant=0;
		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;
}