zzuli 1990 985的“树”难题

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

YES

2 1 0

1 1 2

NO

要用都邻接表，我用的是-链式向前星。

加上并查集，dfs其实就是从根节点把邻接表跑一遍。

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <stack>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;

#define LL long long
#define N 10010
#define mod 1000000007
struct node
{
    int to;
    int w;
    int next;
}a[N];
int head[N],v[N],s[N],f[N],t;
void add(int to,int w)
{
    a[t].to=to;
    a[t].w=w;
    a[t].next=head[to];
    head[to]=t++;
}//以上是邻接表（静态建表，链式向前星）
int findd(int x)
{
    if(x==f[x])
        return x;
    else
    {
        s[f[x]]++;
        return findd(f[x]);
    }
}//并查集
void dfs(int to)
{
    for(int i=head[to];i!=-1;i=a[i].next)
    {
        if(!v[a[i].w]&&a[i].w!=f[to])
        {
            f[a[i].w]=to;
            v[a[i].w]=1;
            dfs(a[i].w);
        }
    }
}//遍历一遍把能连起来的标记一下。
int main()
{
    int T,n,root;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&root);
        int to,w,flag=1;
        t=1;
        memset(head,-1,sizeof(head));
        memset(v,0,sizeof(v));
        for(int i=1;i<=n;i++)
            f[i]=i,s[i]=0;
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&to,&w);
            add(to,w);
            add(w,to);//双向
        }
        v[root]=1;
        dfs(root);//从根处开始遍历
        for(int i=1;i<=n;i++)
            if(findd(i)!=root)
            {
                flag=0;break;
            }
        if(flag)
        {
            printf("YES\n");
            for(int i=1;i<=n;i++)
            {
                printf("%d",s[i]);
                if(i==n)printf("\n");
                else printf(" ");
            }
            for(int i=1;i<=n;i++)
            {
                printf("%d",f[i]);
                if(i==n)printf("\n");
                else printf(" ");
            }
        }
        else
            printf("NO\n");
    }
    return 0;
}