【bzoj 1051】[HAOI2006]受欢迎的牛

[HAOI2006]受欢迎的牛

Time Limit: 10 Sec Memory Limit: 162 MB
Submit: 5255 Solved: 2787
[Submit][Status][Discuss]
Description

　　每一头牛的愿望就是变成一头最受欢迎的牛。现在有N头牛，给你M对整数(A,B)，表示牛A认为牛B受欢迎。 这
种关系是具有传递性的，如果A认为B受欢迎，B认为C受欢迎，那么牛A也认为牛C受欢迎。你的任务是求出有多少头
牛被所有的牛认为是受欢迎的。
Input

　　第一行两个数N,M。 接下来M行，每行两个数A,B，意思是A认为B是受欢迎的（给出的信息有可能重复，即有可
能出现多个A,B）
Output

　　一个数，即有多少头牛被所有的牛认为是受欢迎的。

Sample Input

3 3

1 2

2 1

2 3
Sample Output

1
结论不再赘述

#include<bits/stdc++.h>
using namespace std;
const int maxn = 10005;
const int maxm = 50005;
int h1[maxn],nx1[maxm],to1[maxm],cnt1;
int h2[maxn],nx2[maxm],to2[maxm],cnt2;
int scc[maxn],low[maxn],pre[maxn],sz[maxn],scccnt,tim,ans;
int n,m;
stack<int> sta;
void addEdge1(int u,int v){ to1[++cnt1]=v;nx1[cnt1]=h1[u];h1[u]=cnt1;}
void addEdge2(int u,int v){ to2[++cnt2]=v;nx2[cnt2]=h2[u];h2[u]=cnt2;}
void dfs(int nd){
    pre[nd]=low[nd]=++tim;sta.push(nd);
    for(int i=h1[nd];i;i=nx1[i]){
        if(pre[to1[i]]==0){
            dfs(to1[i]);low[nd]=min(low[nd],low[to1[i]]);
        }else if(scc[to1[i]]==0){
            low[nd]=min(low[nd],pre[to1[i]]);
        }
    }
    if(low[nd]==pre[nd]){
        int x;scccnt++;
        do{
            x=sta.top();sta.pop();
            scc[x]=scccnt;
            sz[scccnt]++;
        }while(x!=nd);
    }
}
void tarjan(){for(int i=1;i<=n;i++)if(pre[i]==0)dfs(i);}
void init(){scanf("%d%d",&n,&m);for(int i=1;i<=m;i++){int a,b;scanf("%d%d",&a,&b);addEdge1(a,b);}tarjan();}
void link(){for(int i=1;i<=n;i++){for(int j=h1[i];j;j=nx1[j]){if(scc[i]!=scc[to1[j]])addEdge2(scc[i],scc[to1[j]]);}}}
void getans(){for(int i=1;i<=scccnt;i++){if(h2[i]==0){if(ans>0){ans=0;return;}else{ans = sz[i];}}}}
int main(){
    init();
    link();
    getans();
    cout<<ans;
    return 0;
}