蓝桥幼儿园（并查集）

蓝桥幼儿园（并查集）

题目描述：

蓝桥幼儿园的学生是如此的天真无邪，以至于对他们来说，朋友的朋友就是自己的朋友。小明是蓝桥幼儿园的老师，这天他决定为学生们举办一个交友活动，活动规则如下：

小明会用红绳连接两名学生，被连中的两个学生将成为朋友。小明想让所有学生都互相成为朋友，但是蓝桥幼儿园的学生实在太多了，他无法用肉眼判断某两个学生是否为朋友。于是他起来了作为编程大师的你，请你帮忙写程序判断某两个学生是否为朋友（默认自己和自己也是朋友）。

输入描述：

第1行包含两个正整数N，M，其中N表示蓝桥幼儿园的学生数量，学生的编号分别为1~ N。之后的第2~M+1行每行输入三个整数，op，x，y：

如果op=1，表示小明用红绳连接了学生x和学生y。

如果op=2，请你回答小明学生x和学生y是否为朋友。

1 <=N,M <= 8 * 10⁵ , 1 <= x, y <= N.

输出描述：

对于每个op=2的输入，如果x和y是朋友，则输出一行 YES，否则输出一行 NO。

输入样例：

5 5
2 1 2
1 1 3
2 1 3
1 2 3
2 1 2

输出样例：

NO
YES
YES

代码：

#include<stdio.h>
int f[11000000];
int n,m;
void init()
{
	for(int i=1;i<=n;i++)
		f[i]=i;
}

int find(int v)
{
	if(f[v]==v)
		return v;
	else 
	{
		f[v]=find(f[v]);
		return f[v];
	}
}

void merge(int v,int u)
{
	int t1,t2;
	t1=find(v);
	t2=find(u);
	if(t1!=t2)
		f[t2]=t1;
	return ;
}

int main()
{
	int x,y,op;
	scanf("%d %d",&n,&m);
	init();
	for(int i=1;i<=m;i++)
	{
		scanf("%d %d %d",&op,&x,&y);
		if(op==1)
			merge(x,y);
		if(op==2)
		{
			if(find(x)==find(y))  //如果两位同学是朋友 
				printf("YES\n");
			else printf("NO\n");
		}
	}
	return 0;
}