并查集的基础思想和题目

南昌理工学院acm集训

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

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

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

在初始化是我们建立一个数组代表每一个元素的所隶属的集合
数组的下标代表元素，数组的内容代表元素所在的集合的根节点，通过根节点的判断来判断两个元素和是否是一个集合内部。

for (int i ;i<=n;i++)//n代表的是集合的元素个数，将每一个元素的根节点都初始化指向自己。
p[i]=i;

集合与集合的合并就用集合的根节点的合并来进行

int find (int a )
{
    while (a!=p[a])  a=p[a];//找到a元素的根节点
    return a;//返回根节点
}

当所合并的集合越来越的的时候，子树所连接的长多会越来越长，在通过find函数进行查询的时候就会消耗大量的计算时间，在进行比较大的运算的时候就需要对代码进行时间上的优化。
利用嵌套的函数循环，让途径的每一个点都直接指向根节点，在第二次遍历的时候就只需要一步直接访问到根节点。
对上述的代码进行路径缩短优化，让后续的查找更加快。

int find (int a )
{
    if(a!=p[a])  p[a]=find (p[a]);//路径优化
    return p[a]; //在得到根节点后一步一步的返回将路径上的每一
    个p[a]都指向这个集合的根节点
}

红色代表进过函数迭代之后得到的新的每一个p[a]都指向了根节点

两个集合的合并

p[find(a)]=p[find(b)] ;//通过优化路径的find函数将两个根节点合并

具体题目代码

#include<iostream>
using namespace std;

int dirt[100020];//用的dirt代表的数组P

int find(int a)//优化后的find函数
{
    if(dirt[a]!=a) dirt[a]=find(dirt[a]);
    return dirt[a];
}

int main(){
    for(int i=1;i<=100020;i++)dirt[i]=i;
    int n,m;
    cin>>n>>m;
    char A;
    int b,c;
    for(int i=1;i<=m;i++){
        cin>>A>>b>>c;
        if(A=='M')
            dirt[find(b)]=find(c);
        else if(find(b)==find(c))
            cout<<"Yes"<<endl;
        else cout<<"No"<<endl;
    }
    return 0;
}

                                             南昌理工学院acm集训队员