PKU POJ 2186 Popular Cows 强连通分量

新注册了博客，正好刚刚开始做强连通专题，就当存个模板了

题目简述：n头奶牛，给出若干个欢迎关系a b，表示a欢迎b，欢迎关系是单向的，但是是可以传递的。另外每个奶牛都是欢迎他自己的。求出被所有的奶牛欢迎的奶牛的数目。

思路：先得出每个强连通分块的规模，再检查输入的每一条边，如果这边连着2个不同的强连通分块，那么被指向的分块才可能是答案。当答案不唯一时则说明至少有两头牛没有任何关系，这时输出0.

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int Max = 20009;
#define max(a,b) a>b?a:b

struct node{
    int v,next;
}Edge[100000];

int k,Dindex,Stop,Bcnt;
int head[Max],DFN[Max],LOW[Max],instack[Max],Stap[Max],Belong[Max];
int a[50009],b[50009],cnt[Max];
bool ok[Max];

void add(int u,int v){
    Edge[k].v = v;
    Edge[k].next = head[u];
    head[u] = k++;
}

void tarjan(int i)
{
    int j;
    DFN[i]=LOW[i]=++Dindex;
    instack[i]=true;
    Stap[++Stop]=i;
    for (int w=head[i];w!=-1;w=Edge[w].next)
    {
        j=Edge[w].v;
        if (!DFN[j])
        {
            tarjan(j);
            if (LOW[j]<LOW[i])
                LOW[i]=LOW[j];
        }
        else if (instack[j] && DFN[j]<LOW[i])
            LOW[i]=DFN[j];
    }
    if (DFN[i]==LOW[i])
    {
        Bcnt++;
        do
        {
            j=Stap[Stop--];
            instack[j]=false;
            Belong[j]=Bcnt;
        }
        while (j!=i);
    }
}

int main() {
    int n,m;
    while(scanf("%d%d",&n,&m) == 2){
        memset(head,-1,sizeof(head));
    memset(DFN,0,sizeof(DFN));
    memset(instack,0,sizeof(instack));
    memset(LOW,0,sizeof(LOW));
        k = 0;
        for(int i=0;i<m;i++){
            scanf("%d%d",&a[i],&b[i]);
            add(a[i],b[i]);
        }
        Stop=Bcnt=Dindex=0;
        for(int i=1;i<=n;i++){
            if(DFN[i] == 0)
                tarjan(i);
        }
        for(int i=1;i<=n;i++)
            cnt[Belong[i]]++;

        int num = 0,ans;
        for(int i=0;i<m;i++){
            if(Belong[a[i]] != Belong[b[i]]){
                ok[Belong[a[i]]] = true;
            }
        }
        for(int i=1;i<=Bcnt;i++){
            if(!ok[i]){
                num++;
                ans = cnt[i];
            }
        }
        if(num != 1)
            printf("0\n");
        else
            printf("%d\n",ans);
    }
    return 0;
}