[P2712]摄像头

原题链接

拓扑排序

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<queue>
#include<vector>
#include<climits>
#include<string>
#include<cstdlib>
#include<ctime>
#define MOD 1000000007
#define LL long long
using namespace std;

int cnt,num[2005],nxt[2005],head[2005],n,i,j,x,y,m,f[2005],rd[2005],t,st[2005],ans,maxn,p;

void add(int p1,int p2)
{
    cnt++;

    num[cnt]=p2;
    nxt[cnt]=head[p1];
    head[p1]=cnt;
}

int main()
{
    scanf("%d",&n);

    for(i=1;i<=n;i++)
    {
        scanf("%d%d",&x,&m);

        maxn=max(maxn,x);
        f[x]=1;

        for(j=1;j<=m;j++)
        {
            scanf("%d",&y);
            add(x,y);
            rd[y]++;
        }
    }

    for(i=1;i<=maxn;i++)
        if(f[i]==1&&rd[i]==0)
        {
            t++;
            st[t]=i;
            ans++;
        }

    while(t>0)
    {
        p=st[t];
        t--;


        for(i=head[p];i;i=nxt[i])
        {
            rd[num[i]]--;

            if(rd[num[i]]==0&&f[num[i]]==1)
            {
                t++;
                ans++;

                st[t]=num[i];
            }
        }
    }

    if(ans<n) printf("%d",n-ans);
    else printf("YES");

    return 0;
}