poj2186 tarjan縮點

題目鏈接poj2186
題意：n個奶牛，m個條邊，每條邊a指向b，代表a奶牛認爲b奶牛是受歡迎的，如果b奶牛認爲c奶牛是受歡迎的，那麼a奶牛認爲c奶牛是受歡迎的。現在要你求受其他所有奶牛歡迎的奶牛個數。

題解，把每一個連通塊縮成一個點，然後記錄每個連通塊的出度點，設出度點個數爲cnt1，
cnt1=0,ans=n;
cnt=1,ans=出度爲0的那個聯通塊的大小。
cnt1>1,ans=0;

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cstdio>
using namespace std;
typedef long long ll;
const int inf=2e9;
const int MAXN=10005;
int val[MAXN],du[MAXN],dfn[MAXN];
struct node{
    int to,Next;
}edge[50005];
int tot;
int timeclock,top,cnt,r;
int low[MAXN],stack[MAXN],instack[MAXN],head[MAXN];
int belong[MAXN],sum[MAXN],size[MAXN];
void add(int u,int v)
{
    edge[tot].to=v;
    edge[tot].Next=head[u];
    head[u]=tot++;
}
void tarjan(int a)           //標準的tarjan代碼
{
    dfn[a]=low[a]=++timeclock;
    instack[a]=1;
    stack[++top]=a;
    for(int i=head[a];i;i=edge[i].Next)
    {
        int v=edge[i].to;
        if(!dfn[v])
        {
            tarjan(v);
            low[a]=min(low[a],low[v]);
        }
        else
        if(instack[v])
        low[a]=min(low[a],dfn[v]);
    }
    if(dfn[a]==low[a])
    {
        cnt++;
        while(stack[top+1]!=a)
        {
            belong[stack[top]]=cnt;
            instack[stack[top]]=0;
            size[cnt]++;
            sum[cnt]=min(sum[cnt],val[stack[top]]);
            top--;
        }
    }
}
int main()
{
    int n,m,u,v;
    tot=1;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        val[i]=inf,du[i]=0,sum[i]=inf;
    for(int i=1;i<=m;i++)
    {
        cin>>u>>v;
        add(u,v);
    }
    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=edge[j].Next)
    if(belong[i]!=belong[edge[j].to])
    {
        du[belong[i]]++;   //記錄出度
    }
    int ans=0,cnt1=0;
    for(int i=1;i<=cnt;i++)
    {
       if(!du[i])
       {
          cnt1++;
          ans+=size[i];
       }
    }
    if(cnt1==0)ans=n;
    else if(cnt1>1)ans=0;
    printf("%d\n",ans);

    return 0;
}