题目
约翰正在玩积木。有N个block (1 <= N <= 30000)编号为1…N .初始有N个桩,每桩包含一个block。然后John做一些操作P次(1 <= P <= 1000000)。有两种操作:
M X Y:将包含X块的整桩放到包含Y块的桩上。如果X和Y在同一桩中,则忽略此命令。
C X:计算X块下的块数
要求您找出每个C操作的输出。
输入
第一行包含整数P,然后是P行,每一行都包含上面描述的操作。
输出
在一行中输出每个C操作的计数。
题解:不仅需要cnt[ ]数组存储答案,还需要sum[ ]数组维护。合并的时候需要用到。
ans[ra]=sum[rb],sum[rb]+=sum[ra]。
#include <cstdio>
using namespace std;
const int N=300000+5;
int f[N],ans[N],sum[N];
int seek(int x){
if(x==f[x]) return x;
int fa=f[x];
f[x]=seek(f[x]),ans[x]+=ans[fa];
return f[x];
}
int main(){
int n;
while(~scanf("%d",&n)){
getchar();
for(int i=1;i<=n;i++) f[i]=i,ans[i]=0,sum[i]=1;
while(n--){
char ch;scanf("%c",&ch);
if(ch=='M'){
int a,b;scanf("%d%d",&a,&b);++a,++b;
int ra=seek(a),rb=seek(b);
if(ra!=rb) f[ra]=rb,ans[ra]=sum[rb],sum[rb]+=sum[ra];
}
else{
int x;scanf("%d",&x);++x,seek(x);
printf("%d\n",ans[x]);
}
}
}
}