[蓝桥杯] 发现环

发现环

题目

小明的实验室有N台电脑，编号1~N。原本这N台电脑之间有N-1条数据链接相连，恰好构成一个树形网络。在树形网络上，任意两台电脑之间有唯一的路径相连。

不过在最近一次维护网络时，管理员误操作使得某两台电脑之间增加了一条数据链接，于是网络中出现了环路。环路上的电脑由于两两之间不再是只有一条路径，使得这些电脑上的数据传输出现了BUG。

为了恢复正常传输。小明需要找到所有在环路上的电脑，你能帮助他吗？

输入格式

第一行包含一个整数N。

以下N行每行两个整数a和b，表示a和b之间有一条数据链接相连。

对于30%的数据，1 <= N <= 1000
对于100%的数据, 1 <= N <= 100000， 1 <= a, b <= N

输入保证合法。

输出格式

按从小到大的顺序输出在环路上的电脑的编号，中间由一个空格分隔。

输入样例

5
1 2
3 1
2 4
2 5
5 3

输出样例

1 2 3 5

题解

思路

确定图中是否有环：并查集

• 并查集每个集合是一个树结构，没有环路
• 如果一个集合中的结点存在邻边，说明存在环
• 通过并查集确定环路的起点和终点

输出环路

• 根据并查集得到的起点和终点，利用 DFS 输出环路

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int N = 1e5 + 20;
int e[N * 2], ne[N * 2], h[N], idx, n, p[N], S, E, ans[N], cnt; 
bool st[N];

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

int find(int x) {
	if (x != p[x]) p[x] = find(p[x]);
	return p[x];
}

void dfs(int u) {
	
	if (u == E) {
		ans[cnt ++] = E;
		sort(ans, ans + cnt);
		for (int i = 0; i < cnt; i ++) cout << ans[i] << " ";
		return;
	}
	
	st[u] = true;
	ans[cnt ++] = u;
	for (int i = h[u]; i != -1; i = ne[i]) {
		int j = e[i];
		if (!st[j]) 
			dfs(j);	
	}
	cnt --;
	st[u] = false;
}

int main () {
	cin >> n;
	memset(h, -1, sizeof h);
	
	for (int i = 0; i <= n; i ++) p[i] = i;
	
	for (int i = 0; i < n; i ++) {
		int a, b;
		cin >> a >> b;
		
		int pa = find(a), pb = find(b);
		if (pa == pb) 
			S = a, E = b;
		else {
			add(a, b), add(b, a);
			p[pa] = pb;
		}
	}
	
	dfs(S);	
	
	return 0;
}