luogu P1726 上白泽慧音

题目描述

在幻想乡，上白泽慧音是以知识渊博闻名的老师。春雪异变导致人间之里的很多道路都被大雪堵塞，使有的学生不能顺利地到达慧音所在的村庄。因此慧音决定换一个能够聚集最多人数的村庄作为新的教学地点。人间之里由N个村庄（编号为1..N）和M条道路组成，道路分为两种一种为单向通行的，一种为双向通行的，分别用1和2来标记。如果存在由村庄A到达村庄B的通路，那么我们认为可以从村庄A到达村庄B，记为(A,B)。当(A,B)和(B,A)同时满足时，我们认为A,B是绝对连通的，记为<A,B>。绝对连通区域是指一个村庄的集合，在这个集合中任意两个村庄X,Y都满足<X,Y>。现在你的任务是，找出最大的绝对连通区域，并将这个绝对连通区域的村庄按编号依次输出。若存在两个最大的，输出字典序最小的，比如当存在1,3,4和2,5,6这两个最大连通区域时，输出的是1,3,4。

输入输出格式

输入格式：

第1行：两个正整数N,M

第2..M+1行：每行三个正整数a,b,t, t = 1表示存在从村庄a到b的单向道路，t = 2表示村庄a,b之间存在双向通行的道路。保证每条道路只出现一次。

输出格式：

第1行： 1个整数，表示最大的绝对连通区域包含的村庄个数。

第2行：若干个整数，依次输出最大的绝对连通区域所包含的村庄编号。

输入输出样例

输入样例#1：

5 5
1 2 1
1 3 2
2 4 2
5 1 2
3 5 1

输出样例#1：

3
1 3 5

说明

对于60%的数据：N <= 200且M <= 10,000

对于100%的数据：N <= 5,000且M <= 50,000

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <stack>

using namespace std;
const int N = 5010;
const int M = 50010;

int now_ans[N], tmp_ans[N], head[N], low[N], dfn[N];
int n, m, now = 1, ans_js, tmp_js, Tarjan_clock;
bool vis[N];
struct Node{
	int u, v, nxt;
}E[M];
stack <int> S;

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

inline void add(int u, int v)
{
	E[now].u = u;
	E[now].v = v;
	E[now].nxt = head[u];
	head[u] = now ++;
}

bool tmp_ans_is_small_er(int x)
{
	for(int i = 1; i <= x; i ++)
	{
		if(now_ans[i] < tmp_ans[i])
			return 0;
		if(now_ans[i] > tmp_ans[i])
			return 1;
	}
	return 0;
}

void Tarjan(int x)
{
	low[x] = dfn[x] = ++ Tarjan_clock;
	S.push(x);
	vis[x] = 1;
	for(int i = head[x]; ~ i; i = E[i].nxt)
	{
		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(low[x] == dfn[x])
	{
		tmp_js = 0;
		while(x != S.top())
		{
			tmp_ans[++ tmp_js] = S.top();
			vis[S.top()] = 0;
			S.pop();
		}
		tmp_ans[++ tmp_js] = S.top();
		vis[S.top()] = 0;
		S.pop();
		sort(tmp_ans + 1, tmp_ans + tmp_js + 1);
		if(tmp_js > ans_js)
		{
			ans_js = tmp_js;
			for(int i = 1; i <= tmp_js; i ++)
				now_ans[i] = tmp_ans[i];
		}
		else if(tmp_js == ans_js)
			if(tmp_ans_is_small_er(tmp_js))
				for(int i = 1; i <= ans_js; i ++)
					now_ans[i] = tmp_ans[i];
	}
} 

int main()
{
	n = read();
	m = read();
	for(int i = 1; i <= n; i ++)
		head[i] = -1;
	for(int i = 1; i <= m; i ++)
	{
		int u = read();
		int v = read();
		int how = read();
		add(u, v);
		if(how == 2)
			add(v, u);
	}
	for(int i = 1; i <= n; i ++)
		if(!dfn[i])
			Tarjan(i);
	printf("%d
", ans_js);
	for(int i = 1; i <= ans_js; i ++)
		printf("%d ", now_ans[i]);
	return 0;
}
/*
5 5
1 2 1
1 3 2
2 4 2
5 1 2
3 5 1
*/