BZOJ 1051 [HAOI 2006]受欢迎的牛 （图的联通，tarjan）

题目链接：BZOJ 1051

直接用强联通分量暴力缩点，找到出度为0的点。如果有多个出度为零的点，那么就不存在受所有牛欢迎的牛。否则答案就为出度为零的点所对应强联通分量的大小。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<stack>
using namespace std;
#define maxn (10000+10)
int N,M,idx=0,tot=0;
int head[maxn],dfn[maxn],low[maxn],belong[maxn],out[maxn],size[maxn];
bool vis[maxn];
struct node{int v,next;}e[5*maxn];

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

int k=0;
void adde(int u,int v){
	e[k].v=v; e[k].next=head[u]; head[u]=k++;
}

stack<int>q;
void tarjan(int x){
	dfn[x]=low[x]=++idx; vis[x]=1;
	q.push(x);
	for(int i=head[x];i!=-1;i=e[i].next){
		int v=e[i].v;
		if(!dfn[v]){
			tarjan(v); low[x]=min(low[x],low[v]);
		}
		else if(vis[v])low[x]=min(low[x],dfn[v]);
	}
	if(dfn[x]==low[x]){
		tot++; int now=0,t=0;
		while(now!=x){
			now=q.top(); q.pop(); t++;
			belong[now]=tot; vis[now]=0;
		}
		size[tot]=t;
	}
}

int main(){
	memset(head,-1,sizeof(head));
	N=read(); M=read();
	for(int i=1;i<=M;i++){
		int x=read(),y=read();
		adde(x,y);
	}
	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!=-1;j=e[j].next){
			int v=e[j].v;
			if(belong[i]!=belong[v])out[belong[i]]++;
		}
	}
	int cnt=0,ans=0;
	for(int i=1;i<=tot;i++){
		if(!out[i]){ans+=size[i];cnt++;}
	}
	if(cnt>1)ans=0;
	printf("%d\n",ans);
	return 0;
}