蓝桥杯真题—发现环（并查集+DFS）

题目描述

小明的实验室有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。
输入保证合法。

输出描述

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

解题思路

用并查集找到环路上的一点，从这个点开始dfs，若过程中找到原点则回溯并记录经过的点。

代码

n = int(input())
f = [i for i in range(n+1)]
g = [[] for i in range(n+1)]
s = 0
ans =[]
vis = [False for _ in range(n+1)]
def find(x):
    if f[x] != x:
        f[x] = find(f[x])
    return f[x]
def dfs(x):
    for i in g[x]:
        if not vis[i]:
            vis[i] = True
            if i == s or dfs(i):
                ans.append(i)
                return True
    return False
for i in range(n):
    a, b = map(int, input().split())
    g[a].append(b)
    g[b].append(a)
    fa, fb = find(a), find(b)
    if fa == fb:
        s = a
    f[fa] = fb
dfs(s)
ans.sort()
for i in ans:
    print(i, end=' ')