Description
约翰和贝茜在玩一个方块游戏.编号为1到n的n(1≤n≤30000)个方块正放在地上.每个构成一个立方柱.
游戏开始后,约翰会给贝茜发出P(1≤P≤100000)个指令.指令有两种:
1.移动(M):将包含X的立方柱移动到包含Y的立方柱上.
2.统计©:统计名含X的立方柱中,在X下方的方块数目.
写个程序帮贝茜完成游戏.
Solution
一直不会写路径压缩的带权并查集。。
我们记录dis[x]表示x下面有多少个,size[x]表示x所在集合有多少个,find的时候更新dis[],合并的时候更新最浅的dis和size就可以了
Code
#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
const int N=60005;
int fa[N],dis[N],size[N];
int read() {
int x=0,v=1; char ch=getchar();
for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
return x*v;
}
int find(int x) {
if (!fa[x]) return x;
int tmp=find(fa[x]);
dis[x]+=dis[fa[x]];
return fa[x]=tmp;
}
bool merge(int x,int y) {
x=find(x),y=find(y);
if (x==y) return false;
fa[x]=y;
dis[x]=size[y];
size[y]+=size[x];
return true;
}
int main(void) {
rep(i,1,N-1) size[i]=1;
for (int T=read(),opt,x,y;T--;) {
for (opt=getchar();opt!='M'&&opt!='C';opt=getchar());
x=read();
if (opt=='M') {
y=read();
merge(x,y);
} else {
find(x);
printf("%d\n", dis[x]);
}
}
return 0;
}