并查集

1.并查集模板  P3367 【模板】并查集 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

#include <bits/stdc++.h>
using namespace std;

const int N = 2e5 + 10;
int n,m,x,y,z;
int p[N]; //p[x] 是 x 点的祖先

int find(int x)  // 并查集
{
    if (p[x] != x) p[x] = find(p[x]); //父亲不是自己就继续接着找
    return p[x]; //返回祖先
}


int main ()
{
    cin >> n >> m;
    for (int i = 1; i <= n; i ++ ) p[i] = i;
    
    while (m -- )
    {
        cin >> z >> x >> y;
        int a = find(x), b = find(y);
        
        if(z == 1) p[a] = b;//x点的祖先的爹记为y点的祖先
        
        if(z == 2) 
        {
            if(a == b) puts("Y"); // 爹一样就说明在一个集合里
            else puts("N"); //爹不一样说明不在一个集合里
        }
    }
    
    return 0;
}

 acwing836 裸并查集模板

#include<bits/stdc++.h>
using namespace std;

const int N = 100010;
int p[N]; //表示每个元素存的父节点是谁
int n,m;

int find(int x) //返回x的祖宗节点的编号 + 路径压缩
{
    if(p[x] != x) p[x] = find(p[x]); 
    return p[x];
}

int main()
{
    scanf("%d%d", &n, &m);
    
    for (int i = 1; i <= n; i ++ ) p[i] = i; //初始化把每个节点的父节点赋成自己
    
    while (m -- )
    {
       char op[2];
       int a,b;
       scanf("%s%d%d", op, &a,&b);
       
       if(op[0] == 'M') p[find(a)] = find(b);
       else 
       {
           if(find(a) == find(b)) puts("Yes");
           else puts("No");
       }
    }
}

 2.连通块中 点的数量 维护并查集中一个集合的点的数量

#include<bits/stdc++.h>
using namespace std;

const int N = 100010;
int p[N];//表示每个元素存的父节点是谁
int siz[N]; //表示每个集合中点的数量
int n,m;

int find(int x) //返回x的祖宗节点的编号 + 路径压缩
{
    if(p[x] != x) p[x] = find(p[x]); //如果这个点的父节点不是自己那么则向上走一层知道找到根节点等于自己，并且回溯压缩路径
    return p[x];
}

int main()
{
    scanf("%d%d", &n, &m);
    
    for (int i = 1; i <= n; i ++ ) 
    {
        p[i] = i; //初始化把每个节点的父节点赋成自己
        siz[i] = 1;
    }
    
    
    while (m -- )
    {
       char op[2];
       int a,b;
       scanf("%s", op);
       
       if(op[0] == 'C')
       {
           scanf("%d%d", &a, &b);
           if(find(a) == find(b)) continue;
           siz[find(b)] += siz[find(a)];
           p[find(a)] = find(b); //连接其实就是集合的合并
       }
       
       else if(op[1] == '1') //查询是否在同一个集合里面
       {
           scanf("%d%d", &a, &b);
           if(find(a) == find(b)) puts("Yes");
           else puts("No");
       }
       else 
       {
           scanf("%d", &a);
           printf("%d\n",siz[find(a)]);
       }
    }
}