交作业（解法二）

Problem : 交作业

Time Limit: 2 Sec Memory Limit: 128 MB

Description

又到了交作业的时间了,都要找wyf学霸抄作业,但wyf只会借作业给朋友,幸好朋友会借给朋友的朋友.但不是朋友就不会分享,问最后谁写了作业,朋友是相互的,wyf编号是1

Input

第一行n,m,学生数和m对朋友，下来m行,ai和bi是朋友

Output

谁写了作业

Sample Input

10 9
1 4
1 8
1 7
2 3
4 8
5 6
6 7
6 8
9 10

Sample Output

1 4 5 6 7 8

HINT

题解：

由于本蒟蒻的Dijkstra解法过于恶心，所以今天特来发布一篇正常的并查集题解。
题意不多分析，直接进入讲解。

1.思路：

本题用并查集的方法，就是每输入两个数，先判断是否在同一集合内，否，则合并。最后查找答案时，1 to n来找出那些数与1在同一集合内。
但我的思路就错在，盲目地想把每一个与1连通的数的祖先都设为1。。。
但实际上，只需要find(1)==find(x)就可以了。

AC代码如下：

#include <cstdio>
#define r register //卡常数，不要在意
#pragma GCC optimize(2)
using namespace std;

int n,m,a,b;
int fa[1001];

struct union_find_set
{
	void begin(int n)
	{
		for (int i=1;i<=n;i++) fa[i]=i;
	}
	int find(int x)
	{
		return x==fa[x]? x:fa[x]=find(fa[x]);
	}
	void merge(int a,int b)
	{
		int x=find(a),y=find(b);
		if (x!=y) fa[x]=y; //判断是否在同一集合内
	}
} ufs; //结构体定义的函数[跟zc神犇学的]

int main()
{
	scanf("%d%d",&n,&m);
	ufs.begin(n); //初始化
	for (r int i=0;i<m;i++)
	{
		scanf("%d%d",&a,&b);
		ufs.merge(a,b); //将两数进行合并
	}
	int wyf=ufs.find(1);
	for (r int i=1;i<=n;i++)
		if (wyf==ufs.find(i)) printf("%d ",i);
		//若两人的祖先相同，则这个人能够写完作业
	return 0;
}

不当之处请各位大佬及时指出！