并查集 AcWing 837. 连通块中点的数量
原题链接
算法标签
并查集
思路
见代码
代码
#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;
// p存储节点祖宗节点 s存储所属连通块节点数
int p[N], s[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;
// 节点数初始化为1
s[i]=1;
}
while(m--){
char op[5];
scanf("%s", op);
if(op[0]=='C'){
int a=read(), b=read();
if(find(a)!=find(b)){
s[find(b)]+=s[find(a)];
p[find(a)]=find(b);
}
}else if(op[0]=='Q'&&op[1]=='1'){
int a=read(), b=read();
if(find(a)==find(b)){
puts("Yes");
}else{
puts("No");
}
}else{
int a=read();
printf("%lld\n", s[find(a)]);
}
}
return 0;
}
原创不易
转载请标明出处
如果对你有所帮助 别忘啦点赞支持哈