题目大意:给出n,表示有n个立方体,p次操作(p未给出)操作分两种,M a b 将a所在列的正方体整个移动至b所在的立方体上面,C a 计算在a下面有几个立方体。
解题思路:带权并查集,根为最底下的立方体,权值代表当前立方体下面有几个,然后在一个数组记录说每一列有多少个立方体,在和并两列的时候作为移动那列的权值。
#include <stdio.h>
#include <string.h>
const int N = 30005;
int n, f[N], s[N], v[N];
int getfar(int x) {
if (x != f[x]) {
int t = f[x];
f[x] = getfar(f[x]);
v[x] += v[t];
}
return f[x];
}
void init () {
scanf ("%d", &n);
for (int i = 0; i < N; i++) {
f[i] = i;
s[i] = 1;
}
memset(v, 0, sizeof(v));
}
int main () {
init ();
int a, b;
char str[10];
for (int i = 0; i < n; i++) {
scanf("%s", str);
if (str[0] == 'M') {
scanf("%d%d", &a, &b);
int p = getfar(a), q = getfar(b);
if (p != q) {
f[p] = q;
v[p] = s[q];
s[q] += s[p];
}
} else {
scanf("%d", &a);
int p = getfar(a);
printf("%d\n", v[a]);
}
}
return 0;
}