tarjan求调
查看原帖
tarjan求调
1013142
Chizuru_Ichinose楼主2024/3/21 19:40
#include<bits/stdc++.h>
#define int long long
#define maxn (10005)
using namespace std;
int n,m,head[maxn],cnt=1;
int dfn[maxn],low[maxn],st[maxn],col[maxn];
bool vis[maxn];
int temp,top,tot;
int out[maxn];
int x[5*maxn],y[5*maxn];
void intt(void)
{
	for(int i=0;i<=maxn;i++) head[i]=-1;
}
struct node
{
	int next;
	int to;
}edge[5*maxn];
void add_edge(int u,int v)
{	
	edge[cnt].to=v;
	edge[cnt].next=head[u];
	head[u]=cnt++;
}
void tarjan(int index)
{
	dfn[index]=low[index]=tot++;
	vis[index]=1;
	st[++top]=index;
	for(int i=head[index];i!=-1;i=edge[i].next)
	{
		int to=edge[i].to;
		if(!dfn[to])
		{
			tarjan(to);
			low[index]=min(low[index],low[to]);
		}
		else if(vis[to]) low[index]=min(low[index],dfn[to]);
	}
	if(low[index]==dfn[index])
	{
		temp++;
		while(st[top+1]!=index)
		{
			vis[st[top]]=0;
			col[st[top--]]=temp;
		}
	}
}
signed main()
{
	ios::sync_with_stdio(0);
	cin.tie(0); cout.tie(0);
	intt();
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		int in,to;
		cin>>in>>to;
		add_edge(in,to);
		x[i]=in;
		y[i]=to;
	}
	// for(int i=1;i<=n;i++)
	// {
		// for(int j=head[i];j!=-1;j=edge[j].next)
		// {
			// cout<<i<<" "<<edge[j].to<<endl;
		// }
	// }
	for(int i=1;i<=n;i++) if(!dfn[i]) tarjan(i);
	for(int i=1;i<=m;i++)
	{
		if(col[x[i]]!=col[y[i]]) out[col[x[i]]]++;
	}
	int s=0;
	for(int i=1;i<=temp;i++)
	{
		if(!out[i])
		{
			if(s)
			{
				cout<<0<<endl;
				return 0;
			}
			s=i;
		}
	}
	int ans=0;
	//cout<<temp<<endl;
	for(int i=1;i<=n;i++)
	{
		if(col[i]==s)
		{
			ans++;
		}
	}
	cout<<ans<<endl;
	return 0;	
}
2024/3/21 19:40
加载中...