BZOJ[1051]受欢迎的牛

#include<iostream>
#include<cstdio>
#include<cstring>
#define M 100010
using namespace std;
struct point{
    int next,to;
}e[M];
int n,m,cnt,num,tot,top,sz,ans;
int head[M],dfn[M],low[M],out[M],st[M],col[M],sum[M],a[M],b[M];
bool vis[M];
void add(int from,int to)
{
    e[++num].next=head[from];
    e[num].to=to;
    head[from]=num;
}
void tarjan(int x)
{
    dfn[x]=low[x]=++cnt;
    st[++top]=x;
    vis[x]=true;
    for(int i=head[x];i;i=e[i].next)
    {
        int to=e[i].to;
        if(!dfn[to])
        {
            tarjan(to);
            low[x]=min(low[x],low[to]);
        }
        else if(vis[to]) low[x]=min(low[x],dfn[to]);
    }
    if(dfn[x]==low[x])
    {
        tot++;
        while(st[top+1]!=x)
        {
            col[st[top]]=tot;
            vis[st[top]]=false;
            sum[tot]++;
            top--;
        }
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&a[i],&b[i]);
        add(a[i],b[i]);
    }
    for(int i=1;i<=n;i++)
        if(!dfn[i])
            tarjan(i);
    for(int i=1;i<=m;i++)
        if(col[a[i]]!=col[b[i]])
            out[col[a[i]]]++;
    for(int i=1;i<=tot;i++)
        if(out[i]==0) 
            sz++,ans=sum[i];
    if(sz>1) printf("0");
    else printf("%d",ans);
    return 0;
}