bzoj1051 [HAOI2006]受欢迎的牛

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#define MAXN 100008
using namespace std;
int n,m,hh[MAXN],head[MAXN],vet[MAXN],next[MAXN],tot,dfn[MAXN],low[MAXN],color[MAXN];
int tot1,num[MAXN],str[MAXN],cnt,boo[MAXN],top,belong[MAXN],tim;

void add(int x,int y){
    tot++;
    hh[tot]=x;
    next[tot]=head[x];
    head[x]=tot;
    vet[tot]=y;
}

void tarjan(int u){
    tim++;
    dfn[u]=low[u]=tim;
    boo[u]=color[u]=1;
    str[++top]=u;
    for(int i=head[u];i!=-1;i=next[i]){
        int y=vet[i];
        if(color[y]==0){
            tarjan(y);
            low[u]=min(low[u],low[y]);
        }else{
            if(boo[y])
                low[u]=min(low[u],dfn[y]);
        }
    }
    color[u]=2;
    if(dfn[u]==low[u]){
        cnt++;
        int v;
        do{
            v=str[top--];
            belong[v]=cnt;
            boo[v]=1;
        }while(v!=u&&top>0);
    }
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        head[i]=-1;
    for(int i=1;i<=m;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
    }
    for(int i=1;i<=n;i++)
        if(!dfn[i]){
            tarjan(i);
        }
    for(int i=1;i<=tot;i++){
        if(belong[hh[i]]!=belong[vet[i]]){
            num[belong[hh[i]]]++;
        }
    }
    int ans=0,flag;
    for(int i=1;i<=cnt;i++){
        if(num[i]==0){
            ans++;
            if(ans>1){
                printf("0");
                return 0;
            }
            flag=i;
        }
    }
    if(ans==1){
        int res=0;
        for(int i=1;i<=n;i++)
            if(belong[i]==flag)
                res++;
        printf("%d",res);
    }else{
        printf("0");
    }
}