BZOJ 1051 [HAOI2006]受欢迎的牛

来源:互联网 发布:拉尼娅王后知乎 编辑:程序博客网 时间:2024/06/01 09:05

题目描述 传送门


学习了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;}