问题描述
小明的实验室有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
思路:从任意一点开始深度优先搜索该树,一旦搜索到重复的点,就停止搜索,然后一直回退到该重复点,记录回退过程中的点,即为环上的点。
#include <iostream>
#include <queue>
#define N 100005
using namespace std;
int n,node; //node 存储重复点
int u[N<<1],v[N<<1];
int first[N],next[N<<1];
int vis[N];
priority_queue<int,vector<int>,greater<int> >q;
int dfs(int i, int t) //从第 i 点开始搜,此时时间为 t
{
if(vis[i]) //遇到重复点
{
if(vis[i]==t-2) return 0; //因为图是双向的,排除父结点
node=i; //记录重复点
return 1; //停止搜索,开始回退
}
vis[i]=t;
int k=first[i]; //访问邻接表
while(k!=-1)
{
if(dfs(v[k],t+1))
{
if(node) //未退回到重复点
{
q.push(i);
if(u[k]==node) node=0; //已回退到重复点
}
return 1;
}
k=next[k];
}
return 0;
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
first[i]=-1;
}
for(int i=0;i<n;i++) //建立邻接表
{
cin>>u[i]>>v[i];
next[i]=first[u[i]];
first[u[i]]=i;
u[n+i]=v[i];
v[n+i]=u[i];
next[n+i]=first[u[n+i]];
first[u[n+i]]=n+i;
}
dfs(1,1);
while(!q.empty()) //输出结果
{
int tmp=q.top();
cout<<tmp<<' ';
q.pop();
}
return 0;
}