[BZOJ 1051][HAOI 2006]受欢迎的牛(tarjan缩点)

http://www.lydsy.com:808/JudgeOnline/problem.php?id=1051

唔。。。这题好像在POJ上见过？

比较水的题，很好想出思路。牛和牛之间的关系就像有向图，牛a喜欢牛b相当于建立有向边a->b，然后在这个有向图中，每个强连通分量里的牛们相当于是相互喜欢的，把这个图缩点成DAG，DAG里如果有且仅有一个出度为0的点，则这个点对应强连通分量里的所有牛都是受欢迎的牛，如果没有出度为0的点，当然就没受欢迎的牛了，如果出度为0的点的个数大于1，则每个出度为0的强连通分量里的牛喜欢的粉丝个数就会不到n-1个，也没有受欢迎的牛。

题目有点坑，貌似题面上标的数据范围小了，被坑WA了一次。

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <stdlib.h>

#define MAXE 100050
#define MAXV 500050

using namespace std;

struct edge
{
    int u,v,next;
}edges[MAXE];

int n,m;
int head[MAXV],nCount=0;
int low[MAXV],dfn[MAXV],belong[MAXV],stack[MAXV],top=0;
int cnt=0,tot=0; //tot=强连通分量个数
int num[MAXV],outDegree[MAXV]; //num[i]=第i号强连通分量里的点个数,outDegree=出度
bool visit[MAXV];

void AddEdge(int U,int V)
{
    edges[++nCount].u=U;
    edges[nCount].v=V;
    edges[nCount].next=head[U];
    head[U]=nCount;
}

void tarjan(int u) //tarjan缩点
{
    low[u]=dfn[u]=++cnt;
    stack[++top]=u;
    visit[u]=true;
    for(int p=head[u];p!=-1;p=edges[p].next)
    {
        int v=edges[p].v;
        if(!dfn[v])
        {
            tarjan(v);
            low[u]=min(low[u],low[v]);
        }
        else if(visit[v])
            low[u]=min(low[u],dfn[v]);
    }
    if(dfn[u]==low[u])
    {
        tot++;
        int v=-1;
        while(u!=v)
        {
            v=stack[top--];
            belong[v]=tot;
            num[tot]++;
            visit[v]=false;
        }
    }
}

int main()
{
    memset(head,-1,sizeof(head));
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        AddEdge(u,v);
    }
    for(int i=1;i<=n;i++)
        if(!dfn[i])
            tarjan(i);
    for(int i=1;i<=m;i++)
        if(belong[edges[i].u]!=belong[edges[i].v])
            outDegree[belong[edges[i].u]]++;
    int res=0,ans; //res=出度为0的点的个数,ans=出度为0的强连通分量编号
    for(int i=1;i<=tot;i++)
        if(outDegree[i]==0)
        {
            res++;
            ans=i;
        }
    if(res==1) printf("%d\n",num[ans]);
    else printf("0\n");
    return 0;
}