AcWing 837. 连通块中点的数量
原题链接
简单
作者:
跟着灿哥学切菜
,
2021-01-15 15:44:46
,
所有人可见
,
阅读 215
#include <iostream>
using namespace std;
const int N = 100010;
//此时有个巨坑, 就是不能使用size[]来进行集合中元素个数的数组。
//此集合不能取这个名字。
int p[N], s[N];
int find(int x) {
if (p[x] != x) p[x] = find(p[x]);
return p[x];
}
int main() {
int n, m;
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i ++) {
p[i] = i;
s[i] = 1;
}
while (m --) {
char op[5];
int a, b;
scanf("%s", op);
if (op[0] == 'C') {
scanf("%d%d", &a, &b);
//如果两个点在同一个集合中, 则不需要进行连接
if (find(a) == find(b)) continue;
s[find(b)] += s[find(a)];
p[find(a)] = find(b);
} else if (op[1] == '1') {
scanf("%d%d", &a, &b);
if (find(a) == find(b)) puts("Yes");
else puts("No");
} else {
scanf("%d", &a);
printf("%d\n", s[find(a)]);
}
}
return 0;
}