并查集 AcWing 836. 合并集合

并查集 AcWing 836. 合并集合

原题链接

AcWing 836. 合并集合

算法标签

并查集

思路

并查集作用

对于上述操作， 并查集时间复杂度几乎为O(1)
若用数组模拟

并查集模拟

路径压缩， 便于查找

路径压缩后， 查找时间复杂度接近O（1）

代码

#include<bits/stdc++.h>
#define int long long
#define rep(i, a, b) for(int i=a;i<b;++i)
#define Rep(i, a, b) for(int i=a;i>b;--i)
using namespace std;
const int N = 100005;
// 存储节点父亲节点
int p[N];
inline int read(){
   int s=0,w=1;
   char ch=getchar();
   while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
   while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
   return s*w;
}
void put(int x) {
    if(x<0) putchar('-'),x=-x;
    if(x>=10) put(x/10);
    putchar(x%10^48);
}
int find(int x){
    if(p[x]!=x){
        return p[x]=find(p[x]);
    }else{
        return p[x];
    }
}
signed main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n=read(), m=read();
    // 父节点初始化 
    rep(i, 1, n+1){
        p[i]=i;
    }
    while(m--){
        char op[2];
        scanf("%s", op);
        int a=read(), b=read();
        if(op[0]=='M'){
        	// 插入
            p[find(a)]=find(b);
        }else{
            if(find(a)==find(b)){
                puts("Yes");
            }else{
                puts("No");
            }
        }
    }
    return 0;
}

原创不易
转载请标明出处
如果对你有所帮助 别忘啦点赞支持哈