并查集【模板】(题解)

版权声明:华天才原创 https://blog.csdn.net/weixin_43891469/article/details/85133072

题目描述

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

输入格式

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

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

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

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

输出格式

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

标程

#include<iostream>
#include<cstdio>
using namespace std;
int n,m,ans,f[10010],p1,p2,p3;
int find(int k){
    if(f[k]==k)return k;
    return f[k]=find(f[k]);
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) f[i]=i;
    for(int i=1;i<=m;i++){
        cin>>p1>>p2>>p3;
        if(p1==1) f[find(p2)]=find(p3);
        else
            if(find(p2)==find(p3)) printf("Y\n");
            else printf("N\n");
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_43891469/article/details/85133072