P3367 【模板】并查集
题目描述:
如题,现在有一个并查集,你需要完成合并和查询操作。
输入格式:
第一行包含两个整数 N,MN,M ,表示共有 N 个元素和 M 个操作。
接下来 M 行,每行包含三个整数 Zi,Xi,Yi
当 Zi= =1 时,将 Xi与 Yi所在的集合合并。
当 Zi= =2 时,输出 Xi 与 Yi 是否在同一集合内,是的输出 Y ;否则输出 N 。
输入 :
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
分析:
这道题目是一个非常经典并且简单的并查集,并查集主要的函数就是:
1.找到父节点
2.查询两个节点是否在同一集合内
3.合并两个节点所在的字迹
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int INF=0x3f3f3f3f;
int x[200000+10],y[200000+10],z[200000+10];
int f[200000+10];//存储代表点
int findfather(int x){//找到代表点
if(f[x]==x){
return x;
}
else{
f[x]=findfather(f[x]);
}
return f[x];
}
void isSameSet(int x,int y){//查询
if(findfather(x)==findfather(y)){
cout<<"Y"<<endl;
}
else{
cout<<"N"<<endl;
}
}
void unio(int x,int y){
f[findfather(y)]=findfather(x);
}
int main(){
ios::sync_with_stdio(false);
int n,m;
cin>>n>>m;
for(int i=1;i<=m;i++){
cin>>z[i]>>x[i]>>y[i];
f[i]=i;
}
for(int i=1;i<=m;i++){
if(z[i]==1){//合并
unio(x[i],y[i]);
}
if(z[i]==2){//查询
isSameSet(x[i],y[i]);
}
}
return 0;
}