3367 【模板】并查集

题目描述

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

输入输出格式

输入格式：

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

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

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

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

 

输出格式：

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

 

代码：

#include<iostream>
using namespace std;
int n, m;
int par[10010];
int getpar(int x){
    if(par[x]!=x) par[x] = getpar(par[x]);
    return par[x];
}
void merge(int x, int y){
    if(getpar(x) == getpar(y))
        return;
    par[getpar(x)] = getpar(y);
}
int query(int x, int y){
    return getpar(x)==getpar(y);
}
int main(){
    cin>>n>>m;
    int i;
    for(i = 1; i <= n; i++)
        par[i] = i;
    for(i = 1; i <= m; i++){
        int z, x, y;
        cin>>z>>x>>y;
        if(z == 1){
            merge(x, y);
        }
        else if(z == 2){
            if(query(x,y))
                 cout<<"Y"<<endl;
            else cout<<"N"<<endl;
        }
    }
    return 0;
}

备注：

本来自己写了一遍，然后只得了20分orz

标黄的两行注意一下。

我要去刚Kruscal了。。