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

时间限制: 1Sec 内存限制: 128MB 提交: 778 解决: 299

题目描述
小明的实验室有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<vector>
#include<string>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<stdio.h>
using namespace std;
struct node{
	int a,b;
}P[100005];
vector<int>G[100005],path;
set<int>se;
int visV[100005],visE[100005];
void DFS(int s){
	if(visV[s]==1){
		for(int i=path.size()-1;i>=0;i--){
			se.insert(path[i]);
			if(path[i]==s)break;
		}
		return ;	
	}
	path.push_back(s);
	visV[s]=1;
	for(int i=0;i<G[s].size();i++){
		if(visE[G[s][i]]==0){
			visE[G[s][i]]=1;
			int v=(s==P[G[s][i]].a?P[G[s][i]].b:P[G[s][i]].a);
			DFS(v);
			visE[G[s][i]]=0;
		}
	}
	path.pop_back();
	visV[s]=0;
}
int main(){
	int N; 
	scanf("%d",&N);
	for(int i=0;i<N;i++){
		int a,b;
		scanf("%d %d",&a,&b);
		G[a].push_back(i); 
		G[b].push_back(i);
		P[i].a=a,P[i].b=b;
	}
	DFS(1);
	for(set<int>::iterator it=se.begin();it!=se.end();it++){
		if(it!=se.begin())printf(" ");
		printf("%d",*it);
	}
	return 0;
}