数组模拟并查集

并查集主要能高效实现以下操作

1. 判断两个元素是否属于一个集合
2. 判断某个元素属于哪个集合

数组模拟并查集的思想

1. 数组的角标是当前元素的值，数组存放的值是该角标元素的父亲节点
2. 利用find（）函数递归寻找元素父亲节点的时候，我们将所有元素都指向树的根节点

一道模板题
链接：https://www.acwing.com/problem/content/description/838/

实现

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+10;
int n, m, p[N];

int find(int a){
    if(p[a] != a) p[a] = find(p[a]);  //当p[a]!=a的时候也就是a不是根节点，递归寻找其根节点，并用根节点修改其父节点的值，所以完成后每个集合的元素都指向了其根节点
    return p[a];
}


int main(){
    scanf("%d %d",&n, &m);
    for(int i=1;i<=n;i++) p[i] = i;
    while(m--){
        char op[10];
        int x, y;
        scanf("%s %d %d",op, &x, &y);
        if(op[0] == 'M'){
            if(find(x) != find(y)) p[find(x)] = find(y);
        }
        else{
            if(find(x) == find(y)) printf("Yes\n");
            else printf("No\n");
        }
    }
    return 0;
}