[蓝桥杯][2017年第八届真题]发现环（Python版本）

问题 1841: [蓝桥杯][2017年第八届真题]发现环

时间限制: 1Sec 内存限制: 128MB 提交: 763 解决: 269

题目描述
小明的实验室有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

这道题的解题思路是找到一个环，然后把环中的元素记录下来。这里要用到并查集的概念，我在网上找到一个非常详细的讲解，这里把链接放在这，不懂的可以参照一下博客
并查集
并查集 代表连通分支, 当然, 属于同一个连通分支的节点一定彼此连通

在输入边时, 合并连通分支, 如果两个节点已经连通(即已经属于相同连通分支), 新加入的边与之前的若干条边构成环路, 分别以这两个节点作为搜索的 起点 和 终点

一边搜索一边保存搜索到的节点, 搜索到终点时, 按序输出

N=int(input())
father=[i for i in range(N+1)]
G=[[]for i in range(N+1)]#建立一个空列表
res=[0 for i in range(N)]
vis=[False for i in range(N+1)]
Input=[]
def dfs(x,len1):
    res[len1]=x
    if x==sb:
        for i in range(len1+1):
            Input.append(res[i])
        Input.sort()
        for i in range(len1+1):
            print(Input[i],end=" ")
        return
    vis[x]=True
    for i in range(len(G[x])):
        if vis[G[x][i]]==False:
            dfs(G[x][i],len1+1)
    vis[x]=0
def getFather(x): #找并查集 根节点, 同时路径压缩
    global father
    t=x
    while father[x]!=x:
        x=father[x]#x 已为根, 开始压缩路径
    while t!=x:
        curf=father[t]
        father[t]=x
        t=curf
    return x
for i in range(N):
    a,b=map(int,input().split())
    c,d=getFather(a),getFather(b)
    if c!=d: #如果当前两点没有连通，则把两点进行连通，并建立关系表
        father[d]=c
        G[a].append(b)
        G[b].append(a)
    else: #如果当前两点已经连通则表明这两点必然在环上，记录两点，用于搜索
        sa=a
        sb=b
dfs(sa,0)#从sa为起点搜索到fy结束