BZOJ(3) 1051: [HAOI2006]受欢迎的牛

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAXN 50010
using namespace std;
int n,m,tim,tot,top,sumcol;
int col[MAXN],sum[MAXN],num[MAXN];
int to[MAXN],net[MAXN],from[MAXN],head[MAXN];
int vis[MAXN],low[MAXN],dfn[MAXN],strack[MAXN],visstrack[MAXN];
void add(int u,int v){
    to[++tot]=v;from[tot]=u;net[tot]=head[u];head[u]=tot;
}
void tarjin(int now){
    strack[++top]=now;
    low[now]=dfn[now]=++tim;
    vis[now]=1;visstrack[now]=1;
    for(int i=head[now];i;i=net[i])
        if(visstrack[to[i]])
            low[now]=min(low[now],dfn[to[i]]);
        else if(!vis[to[i]]){
            tarjin(to[i]);
            low[now]=min(low[now],low[to[i]]);
        }
    if(low[now]==dfn[now]){
        sumcol++;sum[sumcol]++;
        col[now]=sumcol;
        while(strack[top]!=now){
            col[strack[top]]=sumcol;
            visstrack[strack[top]]=0;
            top--;sum[sumcol]++;
        }
        visstrack[now]=0;
        top--;
    }
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v);
    }
    for(int i=1;i<=n;i++)
        if(!vis[i])    tarjin(i);
    for(int i=1;i<=tot;i++)
        if(col[from[i]]!=col[to[i]])
            num[col[from[i]]]++;
    int flag=0,ans;
    for(int i=1;i<=sumcol;i++)
        if(num[i]==0){
            flag++;
            ans=i;
        }
    if(flag!=1)    printf("0\n");
    else printf("%d\n",sum[ans]);
    
}