洛谷P3367【模板】并查集

题目描述

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

输入格式

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

接下来M行，每行包含三个整数Z_i，X_i，Y_i。

当 Z_i=1时，将X_i与Y_i所在的集合合并。

当Z_i​=2时，输出X_i​与Y_i​是否在同一集合内，是的输出Y；否则输出N。

输出格式

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

 

 数据范围

1≤n≤10⁴，1≤m≤2×10⁵。

思路：

这道题是并查集的模板题。我们只需要维护好并查集的find和union两个基本操作即可（注意要使用路径压缩，一定要确定好每一个集合的代表元素唯一）。

代码：

#include<cstdio>
#include<iostream>
using namespace std;
int father[10001]; 
int m,n,i,x,y,z;
int find(int x){
    if(father[x]!=x){
        return father[x]=find(father[x]);
    }else{
        return x;
    }
}
void unionn(int r1,int r2){
    father[r2]=r1;
}
int main(){
    cin>>n>>m;
    for(i=1;i<=n;i++){
        father[i]=i;
    }
    for(i=1;i<=m;i++){
        cin>>z>>x>>y;
        if(z==1){
            int r1=find(x);
            int r2=find(y);
            if(r1!=r2){
                unionn(r1,r2);
            }
        }else if(z==2){
            if(find(x)==find(y)){
                cout<<'Y'<<endl;
            }else{
                cout<<'N'<<endl;
            }
        }
    }
    return 0;
}