poj2186Popular Cows tarjan缩点

//n个奶牛，
//A B 表示A认为B出名，而且其有传递性
//如A认为B出名，B认为C出名，那么A认为C出名
//问有多少头奶牛所有的奶牛都认为其出名
//先对这个图缩点，记录每一个缩点的个数，然后找出度为0有且只有一个点
//如果是，那么那个点缩了几个点就是答案
#include<cstdio>
#include<cstring>
#include<iostream>
#include<vector>
using namespace std ;
const int maxn = 100010 ;
int dfn[maxn] , low[maxn] , stack[maxn] ;
int isstack[maxn] , head[maxn] ;int vis[maxn] ;
int belong[maxn] , sum[maxn] , ans[maxn] ;
int step , num , top ;
vector<int> vec[maxn] ;
int n , m ;
void init()
{
    for(int i = 0;i <= n;i++)
    vec[i].clear() ;
    step = num  = top = 0 ;
    memset(dfn , 0 , sizeof(dfn)) ;
    memset(vis ,0 , sizeof(vis)) ;
    memset(isstack , 0 , sizeof(isstack)) ;
    memset(sum , 0 , sizeof(sum)) ;
    memset(ans , 0 , sizeof(ans)) ;
}
void tarjan(int u)
{
    isstack[u] = 1;
    stack[++top] = u ;
    low[u] = dfn[u] = ++step ;
    for(int i = 0;i < vec[u].size() ; i++)
    {
        int v = vec[u][i] ;
        if(!dfn[v])
        {
            tarjan(v) ;
            low[u] = min(low[u] , low[v]) ;
        }
        else if(isstack[v])
        low[u] = min(low[u] , dfn[v]) ;
    }
    if(low[u] == dfn[u])
    {
        num++ ;
        int v = 0 ;
        while(u != v)
        {
            v = stack[top--] ;
            belong[v] = num ;
            ans[num]++ ;
            isstack[v] = 0 ;
        }
    }
}
int main()
{
    while(~scanf("%d%d" , &n , &m))
    {
        init() ;
        int u , v ;
        for(int i = 1;i <= m;i++)
        {
            scanf("%d%d" , &u , &v) ;
            vec[u].push_back(v) ;
        }
        for(int i = 1;i <= n;i++)
        if(!dfn[i])
        tarjan(i);
        int flag = 0 ;
        for(int i = 1;i <= n;i++)
          for(int j = 0 ; j < vec[i].size() ;j++)
          {
              int v = belong[vec[i][j]] ;
              int u = belong[i] ;
              if(v == u)continue ;
              vis[u] = 1 ;
          }
        int pos ;
        for(int i = 1;i <= num;i++)
        if(!vis[i])
        {
            flag++ ;
            pos = i ;
        }
        if(flag == 1)
        cout<<ans[pos]<<endl;
        else {puts("0");}
    }
}