1900: 985的“树”难题
Time Limit: 1 Sec Memory Limit: 128 MBSubmit: 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
题意很清晰 主要的问题没办法判断那个点是父节点,哪个点是子节点,这时候单单使用并查集做做不出来
需要借助邻接表来确定父子关系,这个难点解决就比较容易做了
#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;
}