题目大意
两种操作,把编号为x的那一堆移到编号为y的上方,或求编号为x的积木下方有多少块积木。
分析
类似于银河星球传说
加权并查集,son表示这一堆的积木数量,dis表示该点到堆顶的距离。
很显然可以求出
answer[x]=son[getf(x)]-dis[x]-1
代码
#include <cstdio>
#include <cctype>
#define maxn 30000
using namespace std;
struct ufind{int f,dis,son;}uf[maxn+1];
int getf(int u){
if (uf[u].f==u) return u;
int fu=getf(uf[u].f);
uf[u].dis+=uf[uf[u].f].dis;
return uf[u].f=fu;//压缩祖先
}
void combi(int x,int y){
int fa=getf(x),fb=getf(y);
uf[fb].f=fa; uf[fb].dis=uf[fa].son; //到堆顶的距离更新uf[fa].son+=uf[fb].son;
}
char alp(); int dig();
int main(){
int m=dig(); for (int i=1;i<=maxn;i++) uf[i].f=i,uf[i].son=1;
while (m--){
char c=alp(); int x=dig();
if (c=='C') printf("%d\n",uf[getf(x)].son-uf[x].dis-1);//输出
else combi(x,dig());
}
return 0;
}
char alp(){
char c=getchar();
while (!isalpha(c)) c=getchar();
return c;
}
int dig(){
int ans=0; char c=getchar();
while (!isdigit(c)) c=getchar();
while (isdigit(c)) ans=ans*10+c-48,c=getchar();
return ans;
}