poj2186(Tarjan scc)

链接：点击打开链接

题意：给定N头牛的牛群和M个有序对(A,B),(A,B)表示牛A喜欢B，并且该关系具有传递性，例如(A,B),(B,C)则(A,C)，求被其他所有牛喜欢的牛的总数

代码：

#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>
using namespace std;

const int siz=10005;
int rt,top,sccnum;
int d[siz],deg[siz];
int sta[siz],dfn[siz],low[siz],scc[siz];
vector<int> G[siz];
void tarjan(int u){
    int i,v;
    sta[++top]=u;
    dfn[u]=low[u]=++rt;
    for(i=0;i<G[u].size();i++){
        v=G[u][i];
        if(dfn[v]==0){
            tarjan(v);
            low[u]=min(low[u],low[v]);
        }
        else if(scc[v]==0)
        low[u]=min(low[u],dfn[v]);
    }
    if(low[u]==dfn[u]){
        sccnum++;
        v=-1;
        while(v!=u){
            v=sta[top--];
            scc[v]=sccnum;
            d[sccnum]++;
        }
    }
}
int main(){                                     //tarjan求出scc，缩点后看出度为0的点是否只有一个                            
    int n,m,i,j,u,v;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++)
    G[i].clear();
    for(i=1;i<=m;i++){
        scanf("%d%d",&u,&v);
        G[u].push_back(v);
    }
    memset(d,0,sizeof(d));
    memset(deg,0,sizeof(deg));
    memset(sta,0,sizeof(sta));
    memset(dfn,0,sizeof(dfn));
    memset(scc,0,sizeof(scc));
    rt=top=sccnum=0;
    for(i=1;i<=n;i++){
        if(dfn[i]==0)
        tarjan(i);
    }
    for(i=1;i<=n;i++){
        for(j=0;j<G[i].size();j++){
            u=scc[i],v=scc[G[i][j]];
            if(u==v)
            continue;
            deg[u]++;
        }
    }
    u=v=0;
    for(i=1;i<=sccnum;i++){
        if(deg[i]==0){
            u=d[i];
            v++;
        }
    }
    if(v>=2)
    puts("0");
    else
    printf("%d\n",u);
    return 0;
}