HDU5326-Work (建反图,遍历)

题目:N个员工,给你n-1对关系u,v,表示u是V的上级,问你的u的下级有K个的数量。

#include<stdio.h>
#include<string.h>
const int N=100;
int nxt[N],pnt[N],head[N];
int n,m,e;
int in[N];
void AddEdge(int x,int y)
{
    pnt[e]=y;nxt[e]=head[x];head[x]=e++;
}
void chuli()
{
    int cnt=0,a[N];
    memset(a,0,sizeof(a));
    while(1)
    {
        int f=0;
        for(int u=2;u<=n;u++)
        {
            if(in[u]==0)
            {
                f=1;
                for(int i=head[u];i!=-1;i=nxt[i])
                {
                    int v=pnt[i];
                    a[v]+=a[u]+1;
                    in[v]--;
                }
                in[u]--;
            }
        }
        if(!f)break;
    }
    for(int i=1;i<=n;i++)
        if(a[i]==m)
            cnt++;
    printf("%d\n",cnt);
}
int main()
{
    int u,v;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(in,0,sizeof(in));
        memset(head,-1,sizeof(head));
        e=0;
        for(int i=0;i<n-1;i++)
        {
            scanf("%d%d",&u,&v);
            AddEdge(v,u);
            in[u]++;
        }
        chuli();
    }
    return 0;
}


你可能感兴趣的:(HDU5326-Work (建反图,遍历))