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 举报

相关文章推荐

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

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

【985系列】985的方格难题

1894: 985的方格难题 Time Limit: 1 Sec  Memory Limit: 128 MB Submit: 450  Solved: 99 SubmitStatusWe...

精选:深入理解 Docker 内部原理及网络配置

网络绝对是任何系统的核心,对于容器而言也是如此。Docker 作为目前最火的轻量级容器技术,有很多令人称道的功能,如 Docker 的镜像管理。然而,Docker的网络一直以来都比较薄弱,所以我们有必要深入了解Docker的网络知识,以满足更高的网络需求。

zzuliOJ 1895:985的0-1串难题(二分+打表)

1895: 985的0-1串难题 Time Limit: 1 Sec  Memory Limit: 128 MB Submit: 203  Solved: 54 SubmitStatusWe...

zzuliOJ 1894:985的方格难题(规律)

1894: 985的方格难题 Time Limit: 1 Sec  Memory Limit: 128 MB Submit: 325  Solved: 59 SubmitStatusWeb ...

ZZULIOJ Problem 1893: 985的数学难题 【位运算】

1893: 985的数学难题 Time Limit: 2 Sec  Memory Limit: 128 MB Submit: 161  Solved: 37 Description 985有n个...

【985系列】985的最大和难题

1899: 985的最大和难题 Time Limit: 1 Sec  Memory Limit: 128 MB Submit: 328  Solved: 40 SubmitStatusW...

985的方格难题

Description 985走入了一个n * n的方格地图,他已经知道其中有一个格子是坏的。现在他要从(1, 1)走到(n, n),每次只可以向下或者向右走一步,问他能否到达(n,n)。若不能...

zzuli-蛤玮打扫教室(线段树区间更新)

1877: 蛤玮打扫教室 Time Limit: 2 Sec  Memory Limit: 128 MB Submit: 555  Solved: 132 SubmitStatusWeb ...

zzuli 1210 二叉树(3) 用线索化二叉树和中跟输出两种方法

我先用的是把中跟输出的结果放到数组里,然后输出它的前面的和后面的#include #include typedef struct tree { char c; struct tree *l,*r;...

ZJU 1990 Subway tree systems - 树的最小表示

题目大意:初始时站在树的根节点,若朝着远离根的方向走,记录“0”,接近根的方向走记录“1”。并且树的每一条边只能来回走一次(即向下和返回)。一个合法的01序列可以描述出一棵树的形态。现在给出两个合法的...
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

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