zzuli 1990 985的“树”难题

原创 2016年08月30日 13:25:30

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;
}


版权声明:乌拉拉,乌拉拉,世界正在变大,我也在变大。 --- zoro

相关文章推荐

ZZULI 1902: 985的因子对难题

1902: 985的因子对难题 Time Limit: 2 Sec Memory Limit: 128 MB Submit: 156 Solved: 45SubmitStatusWeb Board...

zzulioj-1900: 985的“树”难题

1900: 985的“树”难题 Time Limit: 1 Sec  Memory Limit: 128 MB Submit: 42  Solved: 13 SubmitStatusWeb ...
  • Cai_Nia
  • Cai_Nia
  • 2016年08月11日 17:40
  • 253

zzulioj 1900(985的“树”难题)

985的“树”难题Time Limit: 1 Sec Memory Limit: 128 MB Submit: 30 Solved: 7SubmitStatusWeb Board Descri...

ZZULI 多校训练(985专场)

Problem G: 985的数字难题Problem F: 985的红绿灯难题Problem E: 985的买饮料难题Problem C: 985的方格难题...

【多校训练】ZZULIOJ 1894 985的方格难题

1894: 985的方格难题 Time Limit: 1 Sec  Memory Limit: 128 MB Submit: 344  Solved: 69 SubmitStatusWeb Bo...
  • A_ice_
  • A_ice_
  • 2016年08月04日 09:29
  • 191

zzulioj 1902( 985的因子对难题)

类似于素数筛选的方法打表,用数组a来记录每个数组出现的次数(下标法) #include #include #include #include int a[1000010]; int main() ...

zzulioj-1899-985的最大和难题【思维】

题目链接:点击打开链接 1899: 985的最大和难题 Time Limit: 1 Sec  Memory Limit: 128 MB Submit: 334  Solved: 44...

ZZULIOJ-1894-985的方格难题(动态规划)

1894: 985的方格难题Time Limit: 1 Sec Memory Limit: 128 MB Submit: 329 Solved: 62SubmitStatusWeb Board ...

ZZULIOJ-1894-985的方格难题(BFS&规律&DP)

1894: 985的方格难题 Time Limit: 1 Sec Memory Limit: 128 MB  Submit: 329 Solved: 62 SubmitStatusWeb Boa...
  • lu_1110
  • lu_1110
  • 2016年08月04日 09:26
  • 150

【985系列】985的方格难题

1894: 985的方格难题 Time Limit: 1 Sec  Memory Limit: 128 MB Submit: 450  Solved: 99 SubmitStatusWe...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:zzuli 1990 985的“树”难题
举报原因:
原因补充:

(最多只允许输入30个字)