poj 1988(并查集)

题意：有30000个木块，编号从1到30000，然后有两种操作M a b，把木块a所在的堆块放到木块b所在的堆块上，操作C a，询问木块a下面有多少个木块。

题解：用一个数组s[i]存木块i所在堆块一共有多少个木块，因为要求木块i下面有多少个木块，所以再添加一个数组dis[i]存木块i到根节点有多少个木块，这样res = s[i]-dis[i]-1。dis数组更新放在寻找根结点递归的后面，因为要先更新父亲的再更新自己的。

#include <stdio.h>
const int N = 30005;
char str[5];
int p, pa
, s
, dis
;

int get_parent(int x) {
if (x != pa[x]) {
int f = pa[x];
pa[x] = get_parent(pa[x]);
dis[x] += dis[f];
}
return pa[x];
}

int main() {
int t;
for (int i = 0; i <= N; i++) {
pa[i] = i;
s[i] = 1;
dis[i] = 0;
}
scanf("%d", &t);
while (t--) {
scanf("%s", str);
if (str[0] == 'M') {
int a, b;
scanf("%d%d", &a, &b);
int px = get_parent(a);
int py = get_parent(b);
pa[py] = px;
dis[py] = s[px];
s[px] += s[py];
}
else {
int a;
scanf("%d", &a);
int px = get_parent(a);
printf("%d\n", s[px] - dis[a] - 1);
}
}
return 0;
}