ch1(数据结构篇)——并查集

1.自然语言描述
把有限的一系列点进行划分；开始各点独立，然后通过点与点之间连线，使得这些点形成>=1个连通分量。核心操作是合并和查询；合并是指将两个集合合并起来形成一个集合，查询是指检查某两个点是否在一个集合内。除了这两个基本操作，并查集还可以维护其他数据，比如每个集合有多少个点，每个点到其根节点的距离等等。

2.代码描述
题目：Acwing.836 合并集合题目链接

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int MAXN=1e5+10;

int n,m,p[MAXN];//p记录每个结点的根结点

int find(int x)
{
    if(p[x]!=x)
        return p[x]=find(p[x]);//这里用到了状态压缩：即不需要只记录每个点的父节点，而是直接记录其根节点，相当于把该点直接连接到根结点上。
    return p[x];
}

int main(void)
{
    cin>>n>>m;
    
    for(int i=1;i<=n;i++)
        p[i]=i;
        
    char op[2];
    for(int i=0;i<m;i++){
        int a,b;
        cin>>op>>a>>b;
        
        int pa=find(a),pb=find(b);
        
        if(op[0]=='M'){
            if(pa!=pb)//根节点不相同，不在一个集合里
                p[pa]=pb;//把a的根连接到b的根上，将a的根节点的根节点换为b的根节点
        }else
            printf("%s\n",pa==pb?"Yes":"No");
        
    }
    
    return 0;
}

题目：Acwing.837 连通块中点的数量题目链接

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>

using namespace std;

const int MAXN=1e5+10;

int n,m,p[MAXN],cnt[MAXN];//cnt用来记录每个点所在的连通块中点的数量

int find(int x)
{
    if(p[x]!=x)
        return p[x]=find(p[x]);
    return p[x];
}

int main(void)
{
    cin>>n>>m;
    
    for(int i=1;i<=n;i++)
        p[i]=i,cnt[i]++;
        
    for(int i=0;i<m;i++){
        char op[3];
        cin>>op;
        
        int a,b;
        if(op[0]=='C'){
            cin>>a>>b;
            int pa=find(a),pb=find(b);
            if(pa!=pb){
                p[pa]=pb;
                cnt[pb]+=cnt[pa];//合并时，连通块中点的数量要相加
            }
        }else if(op[1]=='1'){
            cin>>a>>b;
            int pa=find(a),pb=find(b);
            printf("%s\n",pa==pb?"Yes":"No");
        }else{
            cin>>a;
            cout<<cnt[find(a)]<<endl;
        }
    }
    
    return 0;
}

题目：Acwing.食物链题目链接

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>

using namespace std;

const int MAXN=5e4+10;

int n,m,p[MAXN],d[MAXN];//d保存每个点到其根节点的距离

int find(int x)
{
    if(p[x]!=x){
        int t=find(p[x]);
        d[x]+=d[p[x]];//在状态压缩的同时还要维护d
        p[x]=t;
    }
    return p[x];
}

int main(void)
{
    cin>>n>>m;
    
    for(int i=1;i<=n;i++)
        p[i]=i;
       
    int res=0; 
    for(int i=0;i<m;i++){
        int cmd,a,b;
        
        cin>>cmd>>a>>b;
        
        if(a>n||b>n){
            res++;
            continue;
        }
        
        int pa=find(a),pb=find(b);
        
        //对于本题，利用点到根节点的距离来确定其与根节点的关系:
        //1.d[x]%3==0,与根节点同类
        //2.d[x]%3==1,吃根节点
        //3.d[x]%3==2,被根节点吃
        //这样，三者之间循环牵制的关系就抽象为三者之间的数量关系
        //同类--0，吃根--1，被根吃--2
        if(cmd==1){
            if(pa==pb&&(d[a]-d[b])%3)//不是同类关系，假话
                res++;
            if(pa!=pb){
                p[pa]=pb;
                d[pa]=d[b]-d[a];//合并后为了使(d[a]+d[pa]-d[b])%3仍然成立,在d[a]增加c后( 即(d[a]+c-d[b])%3 )仍保持原来的大小
            }                   //c(也就是d[pa])=d[b]-d[a];
        }else{
            if(pa==pb&&(d[a]-d[b]-1)%3)//不是a吃b
                res++;
            if(pa!=pb){
                p[pa]=pb;
                d[pa]=d[b]+1-d[a];
            }
        }
        
    }
    
    cout<<res<<endl;
    
    return 0;
}