12.5 并查集+暴力DFS——【发现环】

题目描述

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

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

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

输入描述

输出描述

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

输入输出样例

输入：

5
1 2
3 1
2 4
2 5
5 3

输出：

1 2 3 5

---

最终代码

1. c/c++

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+1100;
//容器数组
vector<int> vt[N];
int s[N],vis[N],circle[N];
int start,flag,num;

void init_set(){
    for(int i=1;i<=N;i++)   s[i]=i;
}
int find_set(int x){ //寻找祖宗
    if(x!=s[x])  s[x]=find_set(s[x]);  //路径压缩
    return s[x];
}
void merge_set(int x,int y){
    int tmp = x;
    x = find_set(x);
    y = find_set(y);
    if(x!=y) s[y]=s[x];
    else     start = tmp; //此x和y处于环中,记录一个点
}


void dfs(int x,int step)
{ //从起点start出发找一条回到start的环路
    if(flag)    return;
    
    if(x==start) //绕了一圈回来了，环路结束
        if(vis[x]==1)
        {
            num = step-1;
            flag = 1;
            return ;
        }
        
    circle[step] = x;
    
    
    for(int i=0;i<vt[x].size();i++)   
    {
        int y=vt[x][i];
        if(!vis[y])
        {
            vis[y]=1;
            dfs(y,step+1);
            vis[y]=0;
        }
    }
}
int main(){
   int n;
    scanf("%d",&n);
    init_set();
    for(int i=1;i<=n;i++) 
    {
        int a,b;
        scanf("%d %d",&a,&b);
        //构建二维容器
        vt[a].push_back(b);
        vt[b].push_back(a);
        merge_set(a,b);  //找到环上一个点start
    }
    flag = 0;
    //以start为起点再次搜回来，这就是环
    dfs(start,1);
    
    
    
    sort(circle+1,circle+1+num);
    for(int i=1;i<=num;i++)
        printf("%d ",circle[i]);
    return 0;
}

---

过程理解