P2341 受欢迎的牛

受欢迎的牛

洛谷链接

题目大意：

有n头牛，牛会喜欢其他牛和自己，而喜欢是可以传递的，求被所有牛喜欢的牛的数量。

思路：

这是一道tarjan缩点的题目。被所有牛都喜欢的牛，一定会在一个强连通分量里。所以我们可以先求出所有强连通分量，缩点后找出度为0的点，如果超过1，则说明没有被所有牛喜欢的牛，如果只有1个，那么受欢迎的牛的数量就是这个点所对应的原来的强连通分量中所含的点的个数。

代码：

#include<stack>
#include<cstdio>
#define N 10010
#define M 50010
using namespace std;
stack<int>p;
int next[M],to[M],num,head[N],f[N],dfn[N],low[N],tim,ans2,place[N],n,m,a,b,dis[N],sum,ans;
void add(int false_from,int false_to){
    next[++num]=head[false_from];
    to[num]=false_to;
    head[false_from]=num;
}
void tarjan(int x){
    p.push(x);
    dfn[x]=low[x]=++tim;
    for(int i=head[x];i;i=next[i]){
        if(f[to[i]])
            continue;
        if(dfn[to[i]])
            low[x]=min(low[x],dfn[to[i]]);
        else{
            tarjan(to[i]);
            low[x]=min(low[x],low[to[i]]);
        }
    }
    if(dfn[x]==low[x]){
        ++ans2;
        while(p.top()!=x){
            f[p.top()]=1;
            place[p.top()]=ans2;
            p.pop();
        }
        f[x]=1;
        place[x]=ans2;
        p.pop();
    }
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;++i){
        scanf("%d%d",&a,&b);
        add(a,b);
    }
    for(int i=1;i<=n;++i)
        if(!dfn[i])
            tarjan(i);
    for(int i=1;i<=n;++i)
        for(int j=head[i];j;j=next[j])
            if(place[i]!=place[to[j]])
                dis[place[i]]++;
    for(int i=1;i<=ans2;++i)
        if(!dis[i]){
            sum++;
            ans=i;
        }
    if(sum>1)
        printf("0");
    else{
        sum=0;
        for(int i=1;i<=n;++i)
            if(place[i]==ans)
                sum++;
        printf("%d",sum);
    }
    return 0;
}