BZOJ 1051 [HAOI2006]受欢迎的牛 tarjan

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；

只有1个连通块的时候，很容易想到，全部都指向的奶牛，

一定都在入度为0的那个强连通分量里，

求出入度为0的强连通分量大小即为答案。

是一道不错的模板题……

#include<bits/stdc++.h>
using namespace std;
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;
}
const int 
	N=10005,
	M=50005;
int n,m,Ecnt,Time,top,num;
int DFN[N],LOW[N],stk[N],NUM[N],sum[N];
int X[M],Y[M],out[N];
bool instack[N],vis[N];
struct Edge{
	int next,to;
}E[M<<1];int head[N];
void add(int u,int v){
	E[++Ecnt].next=head[u];
	E[Ecnt].to=v;
	head[u]=Ecnt;
}
void tarjan(int u){
	LOW[u]=DFN[u]=++Time;
	instack[u]=vis[u]=1;
	stk[++top]=u;
	for (int i=head[u];i;i=E[i].next){
		int j=E[i].to;
		if (!vis[j]){
			tarjan(j);
			LOW[u]=min(LOW[u],LOW[j]);
		} else
		if (instack[j]) LOW[u]=min(LOW[u],DFN[j]);
	}
	if (LOW[u]==DFN[u]){
		sum[++num]=0;
		int now=-1;
		for (;now!=u;top--){
			now=stk[top];
			NUM[now]=num,instack[now]=0,
			sum[num]++;
		}
	}
}
void rebuild(){
	Ecnt=0;
	memset(head,0,sizeof(head));
	for (int i=1;i<=m;i++)
		if (NUM[X[i]]!=NUM[Y[i]]) out[NUM[X[i]]]++;
}
int main(){
	n=read(),m=read();
	Ecnt=0;
	for (int i=1;i<=m;i++){
		X[i]=read(),Y[i]=read();
		add(X[i],Y[i]);
	}
	num=0;
	for (int i=1;i<=n;i++)
		if (!vis[i]) top=Time=0,tarjan(i);
	if (num==1){printf("%d\n",n);return 0;}

	rebuild();
	int tmp=0,ans=0;
	for (int i=1;i<=num;i++)
		if (!out[i]) ans+=sum[i],tmp++;
	if (tmp>1) ans=0;
	printf("%d\n",ans);
	return 0;
}