P3367 并查集

Source: https://www.luogu.com.cn/problem/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
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
说明/提示
对于 30% 的数据，N≤10，M≤20 。

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

对于 100% 的数据，1≤N≤10^4, 1≤M≤2×10^5.

思路：并查集

#include <bits/stdc++.h>
using namespace std;

int father[10005];
int Find(int x){
    int t=x;
    while(father[x]!=x){
        x=father[x];
    }
    int ans=x;
    while(father[t]!=t){
        int tmp=father[t];
        father[t]=ans;
        t=tmp;
    }
    return ans;
}
void Union(int x, int y){
    int fax=Find(x);
    int fay=Find(y);
    if(fax!=fay){
        father[fax]=father[fay];
    }
}

int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    int op,a,b;
    for(int i=1;i<=n;i++){
        father[i]=i;
    }
    for(int i=0;i<m;i++){
        scanf("%d%d%d",&op,&a,&b);
        if(op==1){
            Union(a,b);
        }
        else{
            int fax=Find(a);
            int fay=Find(b);
            if(fax==fay){
                printf("Y\n");
            }
            else{
                printf("N\n");
            }
        }
    }
    // system("pause");
    return 0;
}