1900: 985的“树”难题

                1900: 985的“树”难题
    Time Limit: 1 Sec  Memory Limit: 128 MB
            Submit: 34  Solved: 9

    SubmitStatusWeb BoardDescription
    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

确实是水题用并查集判断是不是树然后跑一边DSF就好了

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#include<stack>
using namespace std;
const int MAXX=1003;
struct node{
    int from;
    int to;
    int next;
}edge[MAXX];
int son[MAXX];
int Fa[MAXX];
int n,k;
int F[MAXX];
int head[MAXX];
int ln;
bool K;
bool vis[MAXX];
void init(){
    int i=0;
    for( i = 1 ;i <= n; ++i)
        F[i]=i,Fa[i]=i,son[i]=1;
    memset(head,-1,sizeof(head)); ln=0; K=false;
    memset( vis, false, sizeof(vis) );
}
int Found(int x){
    if( F[x] == x ) return x;
    F[x] = Found( F[x] );
    return F[x];
}
void Union( int x, int y){
    int nx = Found(x);
    int ny = Found(y);
    if(ny!=nx){
        F[ny] = nx;
    }
    else K = true;

}
void join(int u,int v){
    edge[ln].from=u;
    edge[ln].to=v;
    edge[ln].next=head[u];
    head[u]=ln++;
}
void DFS(int u){
    vis[u] = true;
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].to;
        if(!vis[v]){
            DFS(v);
            son[u]+=son[v];
            Fa[v]=u;
        }
    }
}
int main(){
    int T; scanf("%d",&T);
    while(T--){
        scanf("%d %d",&n,&k);
        init();
        int a,b;int i=0;
        for( i = 1; i <= n-1; ++i ){
            scanf("%d %d",&a,&b);
            if( !K ){
                join(a,b);
                join(b,a);
                Union(a,b);
            }
        }
        int t=0;
        for( i = 1;i <= n; ++i){
            if( F[i] == i )
                t++;
        }
        if( t > 1 )
            K=true;
        if( K ) {
            printf("NO\n");
            continue;
        }
        DFS(k);
        printf("YES\n");
        printf("%d",--son[1]);
        for( i = 2; i <= n; ++i )
            printf(" %d",--son[i]);
        printf("\n%d",Fa[1]);
        for( i = 2; i <= n; ++i )
            printf(" %d",Fa[i]);
        printf("\n");
    }
    return 0;
}