洛谷P1793 跑步（并查集）C/C++

P1793 跑步

输入输出样例
输入

6 6
1 2
2 4
2 3
3 5
4 5
5 6

输出

2
2 5

思路：并查集解法，在循环中依次把每个路口都忽略一遍，看结果起点1结点和终点n节点是不是在同一个集合里，如果在，就说明这个忽略的节点不是必经路口；如果不在，就说明这个路口是必经路口，保存下来

AC代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

int f[2021];

int search(int x)
{
	return f[x]==x?f[x]:(f[x]=search(f[x]));
}

int a[8010],b[8010];
int main()
{
	int n,e,s=0,sum[2021];
	cin >> n >> e ;
	for(int i=1;i<=e;i++)
	{
		cin >> a[i] >> b[i] ;
	}
	for(int i=2;i<n;i++)
	{
		for(int j=1;j<=n;j++)
			f[j]=j;
			
		for(int j=1;j<=e;j++){//暴力依次把每个节点忽略一次 
			if(a[j]==i||b[j]==i)continue; 
			int aa=search(f[a[j]]),bb=search(f[b[j]]);
			f[aa]=bb;
		}
		
		if(search(f[1])!=search(f[n])){
			s++;
			sum[s]=i;
		}
	}
	cout << s << endl ;
	for(int i=1;i<=s;i++)
	{
		cout << sum[i] << " " ;
	}
	
	return 0;
}