说明:
学习该模板比较推荐洛谷P3367 【模板】并查集,以下是该题链接及描述:
https://www.luogu.org/problemnew/show/P3367
题目描述
如题,现在有一个并查集,你需要完成合并和查询操作。
输入格式:
第一行包含两个整数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。
提示:
要抓住每个元素的父亲,合理应用路径压缩:查找完根节点后,在递归回来时顺便将路径上的元素的父亲指向根节点。代码不长的。
代码:
#include
#include
#include
#include
#include
#include
#define N 1000001
using namespace std;
int f[N],n,m;
int find(int k)
{
if(k!=f[k]) f[k]=find(f[k]);
return f[k];
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) f[i]=i;
for(int i=1;i<=m;i++)
{
int x,y,z;
scanf("%d%d%d",&z,&x,&y);
x=find(x);
y=find(y);
if(z==1&&x!=y) f[x]=y;
if(z==2)
{
if(x==y) printf("Y\n");
else printf("N\n");
}
}
return 0;
}
相关链接:
C++模板小站:
https://blog.csdn.net/zj_mrz/article/details/80950647
C++快速幂模板:
https://blog.csdn.net/zj_mrz/article/details/80950616
C++高精度减法模板:
https://blog.csdn.net/zj_mrz/article/details/80965324
C++高精度乘法模板:
https://blog.csdn.net/zj_mrz/article/details/80967394