BZOJ1051受歡迎的牛

1051: [HAOI2006]受歡迎的牛
Time Limit: 10 Sec Memory Limit: 162 MB
Submit: 3673 Solved: 1940
Description
每一頭牛的願望就是變成一頭最受歡迎的牛。現在有N頭牛，給你M對整數(A,B)，表示牛A認爲牛B受歡迎。 這種關係是具有傳遞性的，如果A認爲B受歡迎，B認爲C受歡迎，那麼牛A也認爲牛C受歡迎。你的任務是求出有多少頭牛被所有的牛認爲是受歡迎的。
Input
第一行兩個數N,M。 接下來M行，每行兩個數A,B，意思是A認爲B是受歡迎的（給出的信息有可能重複，即有可能出現多個A,B）
Output
一個數，即有多少頭牛被所有的牛認爲是受歡迎的。
Sample Input
3 3
1 2
2 1
2 3
Sample Output
1
HINT
100%的數據N<=10000,M<=50000
tarjan強聯通分量縮點，然後找出度爲0的點，若有就輸出大小，否則無解。。
附上本蒟蒻的代碼：

#include<cstdio>
#include<iostream>
using namespace std;
struct data
{
    int to,next;
}edge[50001],d[50001];
int n,m,cnt,top,dfn[10001],h[10001],low[10001],q[10001],succ,head[10001],belong[10001],have[10001];
bool vis[10001],mark[10001];

int read()
{
    int w=0,c=1; char ch=getchar();
    while (ch<'0' || ch>'9')
      {
        if (ch=='-') c=-1;
        ch=getchar();
      }
    while (ch>='0' && ch<='9')
      w=w*10+ch-'0',ch=getchar();
    return w*c;
}

void dfs(int x)
{
    int now;
    vis[x]=mark[x]=true,low[x]=dfn[x]=++cnt,q[++top]=x;
    int c=head[x];
    while (c)
      {
        if (!vis[edge[c].to]) dfs(edge[c].to),low[x]=min(low[x],low[edge[c].to]);
        else if (mark[edge[c].to]) low[x]=min(low[x],dfn[edge[c].to]);
        c=edge[c].next;
      }
    if (low[x]==dfn[x])
      {
        succ++;
        while (now!=x)
          now=q[top--],mark[now]=0,belong[now]=succ,++have[succ];
      }
}

void rebuild()
{
    cnt=0;
    for (int i=1;i<=n;i++)
      {
        int c=head[i];
        while (c)
          {
            if (belong[i]!=belong[edge[c].to])
              d[++cnt].to=belong[edge[c].to],d[cnt].next=h[belong[i]],h[belong[i]]=cnt;
            c=edge[c].next;
          }
      }
}

void tarjan()
{
    for (int i=1;i<=n;i++) if (!vis[i]) dfs(i);
    rebuild();
}

int main()
{
    int i,x,y,ans=0;
    n=read(),m=read();
    for (i=1;i<=m;i++)
      {
        x=read(),y=read();
        edge[i].to=y,edge[i].next=head[x],head[x]=i;
      }
    tarjan();
    for (i=1;i<=succ;i++)
      if (!h[i])
        if (ans) ans=0;
        else ans=have[i];
    printf("%d",ans);
    return 0;
}