洛谷 P1726 上白泽慧音 邻接表+Tarjan+栈+染色

洛谷   P1726 上白泽慧音   邻接表+Tarjan+栈+染色

Tarjan算法，详见https://blog.csdn.net/mrcrack/article/details/103806804

在线测评地址https://www.luogu.com.cn/problem/P1726

//样例通过，提交AC.2020-1-2 

#include <stdio.h>
#define maxn 5010
#define maxm 50010
int dfn[maxn],low[maxn],tag,head[maxn],tot,n,m,st[maxn],top,vis[maxn],sum,color[maxn],size[maxn];
struct node{
	int to,next;
}e[maxm<<1];
void add_edge(int u,int v){
	tot++,e[tot].to=v,e[tot].next=head[u],head[u]=tot;
}
int min(int a,int b){
	return a<b?a:b;
}
void Tarjan(int u){
	int v,b;
	low[u]=dfn[u]=++tag;
	st[++top]=u,vis[u]=1;
	for(b=head[u];b;b=e[b].next){
		v=e[b].to;
		if(!dfn[v]){
			Tarjan(v);
			low[u]=min(low[u],low[v]);
		}else if(vis[v]) low[u]=min(low[u],low[v]);
	}
	if(dfn[u]==low[u]){
		sum++;
		while(1){
			v=st[top--],color[v]=sum,vis[v]=0,size[sum]++;
			if(v==u)break;
		}
	}
}
int main(){
	int a,b,t,i,mx=-1,j;
	scanf("%d%d",&n,&m);
	for(i=1;i<=m;i++){
		scanf("%d%d%d",&a,&b,&t);
		add_edge(a,b);
		if(t==2)add_edge(b,a);
	}
	for(i=1;i<=n;i++)
		if(!dfn[i])Tarjan(i);
	for(i=1;i<=sum;i++)
		if(mx<size[i])mx=size[i],j=i;
	printf("%d\n",mx);
	for(i=1;i<=n;i++)
		if(color[i]==j)printf("%d ",i);
	return 0;
}