小白算法练习 lanqiao2017 国赛 发现环 tarjan 无向图

时间:2023-02-13 20:40:51
标题:发现环
小明的实验室有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<algorithm>
#include<vector>
#include<stack>
#include<queue>
using namespace std;
const int Max=100008;
vector<int>vec[Max];
stack<int>sta;
int dfn[Max]={0}; //时间戳
int fro[Max]={0}; //最早的连接点
int vis[Max]={0}; 
int belong[Max]={0};
int counts=0;
int scc=0;
void init(int s,int e){
	vec[s].push_back(e);
	vec[e].push_back(s);
}
void tarjan(int s,int f){
	dfn[s]=fro[s]=++counts;
	sta.push(s);
	vis[s]=1;
	for(int i=0;i<vec[s].size();i++)
	{
		int temp=vec[s][i];
		if(temp==f) continue;
		if(vis[temp]==0)
		{
			tarjan(temp,s);
			fro[s]=min(fro[temp],fro[s]);
		}
		else if(vis[temp]==1)
		{
			fro[s]=min(dfn[temp],fro[s]); 
		}
	}
	int v;
	if(fro[s]==dfn[s])
	{
		++scc;
		do{
			v=sta.top();
			sta.pop();
			vis[v]=0;
			belong[v]=scc;
		}while(s!=v);
	}
}
int main(){
	//input & init
	freopen("E://1001.in","r",stdin);
	int N;cin>>N;
	for(int i=1;i<=N;i++)
	{
		int s,e;
		cin>>s>>e;
		init(s,e);
	}
	//tarjan
	tarjan(1,0);
	//output
	int judge[Max];
	int ac=0;
	for(int i=1;i<=N;i++)
	{
		judge[belong[i]]++;
		if(judge[belong[i]]>1)
		{
			ac=belong[i];
			break;
		}
	}
	for(int i=1;i<=N;i++)
	{
		if(belong[i]==ac)
		{
			cout<<i<<" ";	
				cout<<endl;
	return 0;
} 
 一看就是tarjan的模板题,不过我在处理双向边的时候出现了一点小问题,因为当点1-->2的时候2不能再返回1.,示例有点坑,因为当你建单向边的时候结果是对的,因为示例就是一个强连通组成的环。。但单向边肯定是不对的。所以需要将前面的一个节点也作为参数穿进去,只要当前节点!=前面的节点就好了,其他的就真的是一个模板能解决的问题了。