C++算法(Acwing)学习(持续更新中)---并查集

并查集

并查集的应用主要有以下几点：1.合并两个集合 2.查询某个元素的祖宗节点

并查集查找祖宗节点的模板：

int find(int x)
{
    if(p[x]!=x) p[x]=find(p[x]);//每个集合中只有祖宗节点的p[x]值等于他自己,即 p[x]=x;
    return p[x];//找到了便返回祖宗节点的值
}
//注：find 函数不仅有找祖宗的功能，还把这个查找路径上所有节点直接变成了祖宗节点的孩子
//相当于路径压缩

合并两个集合

Acwing836.合并集合

#include<iostream>
using namespace std;

const int N=1e5+10;
int n,m;
int p[N];

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 --)
    {
        char s;
        int a,b;
        cin>>s>>a>>b;
        if(s=='M') p[find(a)]=p[find(b)];
        else 
        {
            if(find(a)!=find(b))
            {
                puts("No");
            }
            else puts("Yes");
        }
    }
    return 0;
}

相关练习

Acwing1250.格子合并

观察题意：发现我们要判断画完线后有没有围成环
而如果能围成一个环，那么最后画的线的两个端点必然已经在一个连通块内
判断是否在一个连通块————使用并查集

算法步骤如下：
如果两个端点已经在同一个集合(find操作)，那么成环
如果两个端点不在同一个集合内(find操作)，那么这两个集合连通，合并两个集合

//注意到这个地图是二维的，而存点时下表是一维的
//所以存点时，我们让点 f[i][j] 对应下标 n * i + j;
//二维转一维 -> (i, j) <=> i * n + j
//注意只有当x,y都从0开始才能用二维转一维的那个公式
int get(int x,int y)
{
    return x*n+y;
}

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

const int N=40010;
int p[N];

int n,m;

//二维数组转化为一维数组
int get(int x,int y)
{
    return x*n+y;
}


int find(int x)//并查集的模板，寻找祖宗节点
{
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}

int main()
{
    cin>>n>>m;
    
    for(int i=0;i<n*n;i++) p[i]=i;//并查集数组初始化
    
    int res=0;
    for(int i=1;i<=m;i++)
    {
        int x,y;
        char d;
        cin>>x>>y>>d;
        x--,y--;//注意只有当x,y都从0开始才能用二维转一维的那个公式
        int a=get(x,y);
        int b;
        if(d=='D') b=get(x+1,y);
        else b=get(x,y+1);
        
        int pa=find(a);//查找a的祖宗节点
        int pb=find(b);//查找b的祖宗节点
        
        if(pa==pb)
        {
            res=i;
            break;
        }
        
        p[pa]=pb;//合并两个集合
    }
    if(!res) puts("draw");
    else cout<<res;
    return 0;
}

Acing 1252 搭配购买