BZOJ 1051 [HAOI2006]受欢迎的牛

---

题目描述 传送门

---

学习了Tarjan缩点。

---

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
#include<stack>
using namespace std;
const int maxn=10005;
vector<int> g[maxn];
int dfs_clock,scc_cnt,sccno[maxn],pre[maxn],lowlink[maxn],cd[maxn],scctot[maxn];
stack<int>s;
void dfs(int u){
    pre[u]=lowlink[u]=++dfs_clock;
    s.push(u);
    for(int i=0;i<g[u].size();i++){
        int v=g[u][i];
        if(!pre[v]){
            dfs(v);
            lowlink[u]=min(lowlink[u],lowlink[v]);
        }
        else if(!sccno[v]){
            lowlink[u]=min(lowlink[u],pre[v]);
        }
    }
    if(lowlink[u]==pre[u]){
        scc_cnt++;
        for(;;){
            int x=s.top();s.pop();  
            sccno[x]=scc_cnt;   
            if(x==u) break;
        }
    }
}
void find_scc(int n){
    dfs_clock=scc_cnt=0;
    memset(sccno,0,sizeof(sccno));
    memset(pre,0,sizeof(pre));
    for(int i=1;i<=n;i++)
        if(!pre[i]) dfs(i);
}
int main(){
    int n,m;
    cin>>n>>m;
    for(int i=0;i<m;i++){
        int a,b;
        scanf("%d%d",&a,&b);
        g[a].push_back(b);
    }
    find_scc(n);
    memset(scctot,0,sizeof(scctot));
    memset(cd,0,sizeof(cd));
    for(int i=1;i<=n;i++){
        scctot[sccno[i]]++;
        for(int j=0;j<g[i].size();j++)
            if(sccno[g[i][j]]!=sccno[i]) cd[sccno[i]]++;
    }
    int ans=233333,tot=0;
    for(int i=1;i<=scc_cnt;i++)
        if(cd[i]==0) ans=scctot[i],tot++;
    if(tot==1) printf("%d\n",ans);
    else puts("0");
    return 0;
}