Acwing 836 合并集合题解

题目：

一共有 n 个数，编号是 1∼n，最开始每个数各自在一个集合中。

现在要进行 m个操作，操作共有两种：

1. M a b，将编号为 a和 b 的两个数所在的集合合并，如果两个数已经在同一个集合中，则忽略这个操作；
2. Q a b，询问编号为 a 和 b 的两个数是否在同一个集合中；

输入格式

第一行输入整数 n 和 m。

接下来 m 行，每行包含一个操作指令，指令为 M a b 或 Q a b 中的一种。

输出格式

对于每个询问指令 Q a b，都要输出一个结果，如果 a和 b在同一集合内，则输出 Yes，否则输出 No。

每个结果占一行。

数据范围

1≤n,m≤100000

---

对于这道题我们可能会想到暴力算法但是这样的话时间效率不高，所以我们用并查集的知识点来解决。

每个集合用一颗树来表示（不是二叉树），每个集合的头节点我们称为祖宗节点，用一个p[N]来存储数的祖宗节点，这样我们去找a,b是否在同一个集合的时候只需要判断他们的祖宗节点是否一样，相同就说明他们在同一个集合。

根据题意，每个集合刚开始只有一个数，则这个数的祖宗节点等于它本身。合并集合的时候，那么我们只需令某一个数的祖宗节点等于另一个数的祖宗节点。

举个例子，p[5]=5,[9]=9;

合并集合：p[5]=9,p[9]=9 （或者p[5]=5,p[9]=5 都是可以的）

所以5的祖宗节点便成了9

此时以9为祖宗节点的集合为{5，9}

#include<iostream>
using namespace std;
const int N=1e+5;
int n,m;
int p[N];
int find(int x)
{
    // ，经上述可发现每个集合的数只有祖宗节点的p[x]值等于他本身
    if(p[x]!=x)
    {
        p[x]=find(p[x]);
    }
    return p[x];
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=0;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); //令a的祖宗节点等于b的祖宗节点，所以需要先找到a的祖宗节点
        }
        else if(op[0]=='Q')
        {
            if(find(a)==find(b))//如果a,b的祖宗节点相同
            {
                cout<<"Yes"<<endl;
            }
            else{
                cout<<"No"<<endl;
            }
        }
    }
    return 0;
}