题目链接:http://www.lydsy.com/JudgeOnline/problem.php?id=3376
题目大意:
编号为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<cstdio> #include<cstdlib> #include<cstring> #include<algorithm> #include<iostream> using namespace std; #define maxn 30100 int 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(" %c%d",&c,&x); if (c=='C') { int ls=d[ffind(x)]; int oz=ffind(ls);//更新 printf("%d ",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; }