题目描述
如题,现在有一个并查集,你需要完成合并和查询操作。
输入格式
第一行包含两个整数N、M,表示共有N个元素和M个操作。
接下来M行,每行包含三个整数Zi、Xi、Yi
当Zi=1时,将Xi与Yi所在的集合合并
当Zi=2时,输出Xi与Yi是否在同一集合内,是的话输出Y;否则话输出N
输出格式
如上,对于每一个Zi=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
说明/提示
时空限制:1000ms,128M
数据规模:
对于30%的数据,N<=10,M<=20;
对于70%的数据,N<=100,M<=1000;
对于100%的数据,N<=10000,M<=200000。
特别注意:fa数组需要进行初始化,每个结点的祖先都是自己
AC代码兼模板
#include <iostream> using namespace std; const int NN = 1e6+10; int N,M; int fa[NN]; int find(int x){ //查询,并压缩路径 if(x != fa[x]) fa[x] = find(fa[x]); return fa[x]; } void join(int x,int y){ //合并x,y所在的集合 int fx = find(x),fy = find(y); if(fx != fy) fa[fx] = fy; } int main(){ cin>>N>>M; for(int i = 1;i<=N;i++) fa[i] = i;//初始化每个结点的祖先是自己 int op,t1,t2; while(M--){ scanf("%d%d%d",&op,&t1,&t2); if(op == 1){ join(t1,t2); //合并 }else{ if(find(t1) == find(t2)){ //查询是否是同一个祖先 cout<<"Y"<<endl; }else{ cout<<"N"<<endl; } } } }