AcWing1174 受欢迎的牛（tarjan缩点）

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+10;
int id[N];
int h[N],ne[N],w[N],e[N],idx;
stack<int> q;
int scnt;
int cnt[N];
int times;
int n,m;
int dfn[N],low[N];
int ins[N];
int in[N];
void tarjan(int u){
    dfn[u]=low[u]=++times;
    int i;
    q.push(u),ins[u]=1;
    for(i=h[u];i!=-1;i=ne[i]){
        int j=e[i];
        if(!dfn[j]){
            tarjan(j);
            low[u]=min(low[u],low[j]);
        }
        else if(ins[j]){
           low[u]=min(low[u],dfn[j]); 
        }
        
    }
    if(dfn[u]==low[u]){
        ++scnt;
        while(1){
            int t=q.top();
            q.pop();
            ins[t]=0;
            cnt[scnt]++;
            id[t]=scnt;
            if(t==u)
            break;
        }
    }
}
void add(int a,int b){
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
int main(){
    cin>>n>>m;
    int i;
    memset(h,-1,sizeof h);
    for(i=1;i<=m;i++){
        int a,b;
        scanf("%d%d",&a,&b);
        add(a,b);
    }
    for(i=1;i<=n;i++){
        if(!dfn[i])
            tarjan(i);
    }
    int j;
    for(i=1;i<=n;i++){
        for(j=h[i];j!=-1;j=ne[j]){
            int k=e[j];
            if(id[i]!=id[k]){
                in[id[i]]++;
            }
        }
    }
    int sum=0;
    int sign=0;
    for(i=1;i<=scnt;i++){
        if(!in[i]){
            sum+=cnt[i];
            sign++;
        }
        if(sign>1){
            sum=0;
            break;
        }
    }
    cout<<sum<<endl;
}