【拓扑排序】蓝桥杯真题 发现环

问题描述

       小明的实验室有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

思路

拓扑排序，可以将不构成环的结点依次去除，最后剩下环

代码

#include<bits/stdc++.h>
#define N 100010
using namespace std;
int node[N]={0}, vis[N]={0}, n;
map<int, vector<int> > edge;
queue<int> que;

//拓扑排序函数 
void _sort(){
	//将只有一条边的结点，不会构成环的边加入队列（少于2条边的节点无法构成环） 
	for(int i=1; i<=n; i++){
		if(node[i]==1){
			que.push(i);
			vis[i]=1;
		}
	}
	
	while(que.size()){
		int t = que.front();
		que.pop();
		
		//因为队列里的结点都只有一条边，所以不需要用到for循环 
		node[edge[t][0]]--;
		if(node[edge[t][0]]==1){
			que.push(edge[t][0]);
			vis[edge[t][0]] = 1;
		}
	}
}

int main(){
	scanf("%d", &n);
	int x, y;
	for(int i=0; i<n; i++){
		scanf("%d%d", &x, &y);
		node[x]++;
		node[y]++;
		edge[x].push_back(y);
		edge[y].push_back(x);
	}
	
	_sort();
	
	//将剩下结点从小到大输出 
	for(int i=1; i<=n; i++){
		if(vis[i]==0){
			printf("%d ",i);
		}
	}
	printf("\n");
	return 0;
}