C++并查集模板

说明:

学习该模板比较推荐洛谷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

转载于:https://www.cnblogs.com/zj-mrz/p/10122458.html

你可能感兴趣的:(C++并查集模板)