zoro_n的博客

只想走得更远

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


阅读更多
版权声明:乌拉拉,乌拉拉,世界正在变大,我也在变大。 --- zoro https://blog.csdn.net/zoro_n/article/details/52368435
个人分类: 图论
上一篇nyoj 1291 不是匹配
下一篇zzuli 1902 985的因子对难题
想对作者说点什么? 我来说一句

[BZOJ 3257]树的难题

树形DP

Horizon_SMZ Horizon_SMZ

2016-03-15 16:43:19

阅读数:1057

没有更多推荐了,返回首页

关闭
关闭