1051: [HAOI2006]受欢迎的牛

题目链接

题目大意：现在有N头牛，给你M对整数(A,B)，表示牛A认为牛B受欢迎。 这
种关系是具有传递性的，如果A认为B受欢迎，B认为C受欢迎，那么牛A也认为牛C受欢迎。求出有多少头牛被所有的牛认为是受欢迎的。

题解：tarjan求SCC,若出度为0的点仅有一个，那么答案即为该点代表的强连通分量中点的个数。否则，答案为0

我的收获：学了tarjan

#include <iostream>
#include <cstdio>
#include <stack>
#include <cstring>
using namespace std;
const int M=100005;

int n,m,tim,t,scnt;
int dfn[M],low[M],head[M],col[M],mark[M],sz[M],h[M];
bool ins[M];

stack <int> s;

struct edge{int fro,to,nex;}e[M*10];

void add(int i,int j){e[t].fro=i,e[t].to=j,e[t].nex=head[i],head[i]=t++;}

void tarjan(int x){
    int now=0;
    dfn[x]=low[x]=++tim;
    s.push(x);ins[x]=1;
    for(int i=head[x];i!=-1;i=e[i].nex){
        int v=e[i].to;
        if(!dfn[v]){tarjan(v);if(low[v]<low[x]) low[x]=low[v];}
        else if(ins[v]&&dfn[v]<low[x]) low[x]=dfn[v];
    }
    if(low[x]==dfn[x]){
        scnt++;
        while(x!=now){
            now=s.top();s.pop();
            ins[now]=0;col[now]=scnt;++sz[scnt];
        }
    }
}

void work()
{
    int tot=0,ans;
    for(int i=1;i<=n;i++)
        if(!dfn[i]) tarjan(i);
    for(int i=0;i<t;i++)//不需要缩点，直接判断即可
        if(col[e[i].fro]!=col[e[i].to])
            h[col[e[i].fro]]++;
    for(int i=1;i<=scnt;i++)
    if(!h[i]) tot++,ans=i;
    if(tot==1) printf("%d\n",sz[ans]); 
    else puts("0");
}

void init()
{
    int x,y;
    cin>>n>>m;t=0;
    memset(head,-1,sizeof(head));
    for(int i=1;i<=m;i++)
        scanf("%d%d",&x,&y),add(x,y); 
}

int main()
{
    init();
    work();
    return 0;
}