AcWing 836. 合并集合

操作：

1.合并两个集合

2.询问两个数是否在同一个集合中

每个集合用一棵树来表示，树根的编号为集合的编号。每个结点存储他的父结点，p[x]表示x的父结点。

将结点的父结点进行存储，每次进行查询时则往往上找，判断是否为根结点，找到所在集合的编号判断所在的集合。

问：1.如何判断树根？

if(p[x]==x)

2.如何求x的集合编号？

while(p[x]!=x) x=p[x];

3.如何合并两个集合？

：将一棵树插到另一棵树上的某个位置上，将被插入的根结点替换为另一棵树的编号。

路径压缩：当一棵树合并到另一棵树上时，将所有结点的父结点都改为另一棵树的编号。

#include <iostream>
using namespace std;
const int N =100010;
int p[N];//定义集合

int fun(int x){
//返回x的祖宗结点 + 路径压缩
    if(p[x]!=x)//当x不是根结点时 就让该结点的父结点等于祖宗结点
        p[x]=fun(p[x]);
    return p[x];
}
int main(){
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=m;i++) p[i]=i;//初始化当前总的集合 每个树根都为自己本身
    while(m--){
        char c[2];//存储选项
        int a,b;//数字
        scanf("%s%d%d",&c,&a,&b);
        if(*c=='M')p[fun(a)]=fun(b);//合并
        else if(fun(a)==fun(b)){
            printf("Yes\n");
        }
        else 
            printf("No\n");
    }
    return 0;
}