题意:
有N堆方块(初始时每堆有一个方块),编号1~N,有两种操作:
M x y:将x所在堆的方块拿起来放到y所在堆的上边。
C x:标号为x的方块下面有多少个方块。
解题思路:暴力肯定会超时,这是一道并查集的题目。每一个方块有三个参数:
par:父亲结点的编号
under:当前结点下面的方块数目,在堆合并和路径压缩中都要更新
sum:如果该结点作为根结点,则表示的是该堆的方块总数
样例演示:
代码:
#include <cstdio>
#include <cstring>
#define M 50000
int par[M], under[M], sum[M];
int get_par(int u) {
if(par[u] != u) {
int tmp = par[u];
par[u] = get_par(par[u]); //更改父亲结点
under[u] += under[tmp]; //更新under参数
}
return par[u];
}
int query(int a, int b) {
return get_par(a) == get_par(b);
}
void merge(int a, int b) {
int t1 = get_par(a);
int t2 = get_par(b);
under[t1] = sum[t2];
sum[t2] += sum[t1];
par[t1] = t2;
}
int main() {
int p;
char cmd[10];
int x, y;
while(~scanf("%d", &p)) {
for(int i=0; i<M; i++) par[i] = i;
memset(under, 0, sizeof(under));
for(int i=0; i<M; i++) sum[i] = 1;
for(int i=0; i<p; i++) {
scanf("%s", cmd);
switch(cmd[0]) {
case 'M':
scanf("%d%d", &x, &y);
merge(x, y);
break;
case 'C':
scanf("%d", &x);
get_par(x);
printf("%d\n", under[x]);
break;
}
}
}
return 0;
}