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
#include
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;
}