蓝桥杯 历届试题 发现环

利用并查集找到存在环的两个点，然后再用dfs搜索路径

http://lx.lanqiao.cn/problem.page?gpid=T453

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <vector>
#include <cstring>
using namespace std;
const int maxn = 1e5+5;
int N;
int a,b;
int s,f;
vector<int>v[maxn]; //二维数组 记录一个点与哪一个点相连
int pre[maxn];   	 //并查集
int vis[maxn];  	  //dfs中标记是否走过某个点
int rid[maxn];   	 //dfs中记录深搜走过的点
int find(int x){
	return x==pre[x] ? x : pre[x]=find(pre[x]);
}
void dfs(int a1,int b1, int n){
	rid[n]=a1;      	//记录走的第n个点是a1
if(a1 == b1){	 		//如果找到b1
sort(rid,rid+n+1); 		//用sort进行排序 从小到大输出
		for(int i=0;i<=n;i++){
			cout<<rid[i];
			if(i==n)
				cout<<endl;
			else
				cout<<" ";
		}
		return;
	}
	vis[a1]=1;   	 //记录走过点a1
	for(int i=0;i<v[a1].size();i++){       //在二维数组中与a1相连的所有点都尝试走一次
		int ve = v[a1][i];
		if(!vis[ve])
			dfs(ve,b1,n+1);
	}
}
 int main(){
 	cin>>N;
 	memset(rid,0,sizeof rid);
 	memset(vis,0,sizeof vis);
 	for(int i = 1; i<= N;i++)
 		pre[i]=i;
 	for(int i = 1; i<= N;i++){
 		cin>>a>>b;
 		int fa = find(a);    
 		int fb = find(b);    //并查集
 		if(fa == fb){        //如果原本a和b 就在并查集中 说明a 和b存在环
 			s = a;
 			f = b;
 			
		 }else{
		 	pre[fa] = fb;
		 	v[a].push_back(b);v[b].push_back(a); //将a和b放入各自对应的二维数组
		 }
	 }
	 dfs(s,f,0);
	 return 0;
 }