题目链接:
题目大意:
编号为1到n的n(1≤n≤30000)个方块正放在地上.每个构成一个立方柱.
有P(1≤P≤100000)个指令.指令有两种:
1.移动(M):将包含X的立方柱移动到包含Y的立方柱上.
2.统计(C):统计名含X的立方柱中,在X下方的方块数目.
题解:
带权并查集
存三个东西,x所在立方柱的最顶端fa[x],x所在立方柱的最底端d[x],x上面有多少个立方柱f[x](下面的图画错了不包含x qwq..画的时候一点感觉都没有)。
那么要求的x下方的数目就可以用f[d[x]]-f[x]来表示。
#include#include #include #include #include using namespace std;#define maxn 30100int fa[maxn],f[maxn],d[maxn];int ffind(int x){ if (fa[x]!=x) { int y=fa[x]; fa[x]=ffind(fa[x]); f[x]=f[x]+f[y]; d[x]=d[y]; }return fa[x];}int main(){ //freopen("cubes.in","r",stdin); //freopen("cubes.out","w",stdout); int q,i,x,y;char c; scanf("%d",&q); for (i=1;i<=30000;i++) fa[i]=d[i]=i,f[i]=0; for (i=1;i<=q;i++) { scanf("\n%c%d",&c,&x); if (c=='C') { int ls=d[ffind(x)]; int oz=ffind(ls);//更新 printf("%d\n",f[ls]-f[x]); }else//把x放到y上 { scanf("%d",&y); int fx=ffind(x),fy=ffind(y); fa[fy]=fx;int ls=ffind(d[fx]);//更新 f[fy]=f[d[fx]]+1; d[fx]=d[fy]; } } return 0;}