【WZOI】（基础题库）第914题 并查集

题目描述:

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

输入格式:

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

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

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

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

输出格式:

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

数据范围:

对于 30% 的数据，N≤10，M≤20。

对于 70% 的数据，N≤100，M≤103。

对于 100% 的数据，1≤N≤104，1≤M≤1×105，1≤Xi,Yi≤N，Zi∈{1,2}。

样例输入:

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

样例输出:

N
Y
N
Y

时间限制: 1000ms
空间限制: 256MB

这道题需要的操作分别为：查询，合并

并查集直接数组模拟即可，无需耗费太多精力。

首先进行初始化操作：

void init(){
	for(int i=1;i<=n;i++)f[i]=i;
    //将所有节点的父亲都初始化为自己
}

如何查询？

因为并查集第x项都记录的是这个节点x的父亲，我们可以一路顺藤摸瓜，当节点y的父亲是自己时，那么节点x的祖先就是节点y

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

如何合并？

将集合a节点x的祖先与集合b节点y的祖先连接起来就行。

void merge(int x,int y){f[find(x)]=find(y);}

模板都给了，代码就很简单了吧。

代码实现：

#include<bits/stdc++.h>
using namespace std;
int n,m,k,f[10000005],x,y;
int find(int x){
	if(f[x]==x)return x;
	return find(f[x]);
}
void merge(int x,int y){f[find(x)]=find(y);}
void init(){
	for(int i=1;i<=n;i++)f[i]=i;
}
int main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	cin>>n>>m;
	init();
	for(int i=1;i<=m;i++){
		cin>>k>>x>>y;
		if(k==2){
			int fx=find(x),fy=find(y);
			if(fx==fy)cout<<"Y\n";
			else cout<<"N\n";
		}
		if(k==1)merge(x,y);
	}
	return 0;
}