洛谷P3367：【模板】并查集

https://www.luogu.org/problemnew/show/P3367

题目描述

如题，现在有一个并查集，你需要完成合并和查询操作。

输入输出格式

输入格式：

 

第一行包含两个整数N、M，表示共有N个元素和M个操作。

接下来M行，每行包含三个整数Zi、Xi、Yi

当Zi=1时，将Xi与Yi所在的集合合并

当Zi=2时，输出Xi与Yi是否在同一集合内，是的话输出Y；否则话输出N

 

输出格式：

 

如上，对于每一个Zi=2的操作，都有一行输出，每行包含一个大写字母，为Y或者N

 

输入输出样例

输入样例#1： 复制

4 7
2 1 2
1 1 2
2 1 2
1 3 4
2 1 4
1 2 3
2 1 4

输出样例#1： 复制

N
Y
N
Y

#include <stdio.h>
#define N 10020
int f[N];
int getf(int v)
{
	if(f[v]==v)
		return v;
	else
	{
		f[v]=getf(f[v]);
		return f[v];
	}
}
void merge(int u, int v)
{
	int t1, t2;
	t1=getf(u);
	t2=getf(v);
	if(t1!=t2)
		f[t2]=t1;
}
int main()
{
	int i, n, m, z, x, y;
	scanf("%d%d", &n, &Am);
	for(i=0; i<=n; i++)
		f[i]=i;
	while(m--)
	{
		scanf("%d%d%d", &z, &x, &y);
		if(z==1)
			merge(x, y);
		else
		{
			if(getf(x)==getf(y))
			//if(f[x]==f[y])
				printf("Y\n");
			else
				printf("N\n");
		}
	}
	return 0;	
}