acwing.1174 受欢迎的牛 tarjan求有向图强连通分量

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <stack>
#include <map>
#include <cmath> 
#include <vector>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int N=1e4+10,M=6e4+10,mod=998244353;

int h[N],e[M],ne[M],idx;
void add(int a,int b){e[idx]=b,ne[idx]=h[a],h[a]=idx++;return;}

int timestamp,scc_cnt,dfn[N],low[N];
int id[N],siz[N],in[N];
int st[N],hh=-1;
bool dis[N]; 

void tarjan(int u)
{
	dfn[u]=low[u]=++timestamp;
	st[++hh]=u;dis[u]=true;
	for(int i=h[u];~i;i=ne[i])
	{
		int j=e[i];
		if(!dfn[j])//第一个加入的timestamp要》0；
		{
			tarjan(j);
			low[u]=min(low[u],low[j]);
		}
		else if(dis[j])low[u]=min(low[u],dfn[j]);
	}
	if(dfn[u]==low[u])
	{
		++scc_cnt;
		int y;
		do{
			y=st[hh--];
			dis[y]=false;
			id[y]=scc_cnt;
			siz[scc_cnt]++;
		}while(y!=u);
	}
}
int main()
{
 	int n,m;
 	scanf("%d%d",&n,&m);
 	memset(h,-1,(n+10)*sizeof int); 
 	for(int i=1;i<=n;i++)add(0,i);// e 把所有点联通； 
 	while(m--)
 	{
 		int a,b;
 		scanf("%d%d",&a,&b);
 		add(a,b);
	 }
	tarjan(0);
	
	for(int i=0;i<=n;i++)
		for(int j=h[i];~j;j=ne[j])
		{
			int t=e[j];
			if(id[i]!=id[t])in[id[i]]++;
		}
	
	int x=0,sum=0;
	for(int i=1;i<=scc_cnt;i++)
		if(!in[i])
		x++,sum+=siz[i];
	if(x>1)sum=0;
	printf("%d",sum);
	return 0;	
}
 
void add_(int &a,ll b){a=((a+b)%mod+mod)%mod;return;}