洛谷-2341 [HAOI2006]受欢迎的牛

题目描述
每头奶牛都梦想成为牛棚里的明星。被所有奶牛喜欢的奶牛就是一头明星奶牛。所有奶
牛都是自恋狂，每头奶牛总是喜欢自己的。奶牛之间的“喜欢”是可以传递的——如果A喜
欢B，B喜欢C，那么A也喜欢C。牛栏里共有N 头奶牛，给定一些奶牛之间的爱慕关系，请你
算出有多少头奶牛可以当明星。
输入输出格式
输入格式：
 第一行：两个用空格分开的整数：N和M
 第二行到第M + 1行：每行两个用空格分开的整数：A和B，表示A喜欢B
输出格式：
 第一行：单独一个整数，表示明星奶牛的数量

输入输出样例
输入样例#1：
3 3
1 2
2 1
2 3

输出样例#1：
1

说明
只有 3 号奶牛可以做明星
【数据范围】
10%的数据N<=20, M<=50
30%的数据N<=1000,M<=20000
70%的数据N<=5000,M<=50000
100%的数据N<=10000,M<=50000

解释：强联通分量缩点，裸题。最后直接统计出度为0的节点里面有多少个原始节点个数就好了，同时如果有两个或以上的节点出度为0，0则答案为0

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
const int maxn = 20005, maxm = 50005;
int Index,pd[maxn],DFN[maxn],LOW[maxn];
int tot=0,color[maxn];
int edge,fir[maxn],Next[maxm],to[maxm];
int sta[maxn],top; //手写栈
int n,m,val[maxn],x[maxm],y[maxm],ans;
void add(int x,int y){
    to[++edge]=y; Next[edge]=fir[x]; fir[x]=edge;
}
void tarjan(int x){
    sta[++top]=x;
    pd[x]=1;
    DFN[x]=LOW[x]= ++Index;
    for(int i=fir[x];i;i=Next[i]){
        int v=to[i];
        if(!DFN[v]){
            tarjan(v);
            LOW[x]=min(LOW[x],LOW[v]);
        }
        else if(pd[v]){
            LOW[x]=min(LOW[x],DFN[v]);
        }
    }
    if(DFN[x]==LOW[x]){
        tot++;
        while(sta[top+1]!=x){
            color[sta[top]]=tot;
            pd[sta[top--]]=0;
        }
    }
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&x[i],&y[i]);
        add(x[i],y[i]);
    }
    for(int i=1;i<=n;i++) if(!DFN[i]) tarjan(i);
    memset(fir,0,sizeof(fir));
    memset(Next,0,sizeof(Next));
    memset(to,0,sizeof(to));
    memset(val,0,sizeof(val));
    edge=0;
    for(int i=1;i<=m;i++){
        if(color[x[i]]!=color[y[i]]){
            add(color[x[i]],color[y[i]]);
            val[color[x[i]]]++;
        }
    }
    int root=0,num=0;
    for(int i=1;i<=tot;i++){
        if(!val[i]){
            root=i;
            num++;
        }
    }
    if(num>1) printf("0\n");
    else{
        for(int i=1;i<=n;i++){
            if(color[i]==root){
                ans++;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}