bzoj 3376: [Usaco2004 Open]Cube Stacking 方块游戏带权并查集_约翰和贝茜在玩一个方块游戏。编号为1…n的n(1≤n≤30000)个方块正放在地上,每个-优快云博客

本文介绍了一种方块游戏，玩家需要根据指令移动方块或统计方块数量。游戏涉及30000个编号的立方柱，处理100000个指令。解决方案采用带权并查集数据结构来高效地解决问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意

约翰和贝茜在玩一个方块游戏．编号为1到n的n(1≤n≤30000)个方块正放在地上．每个构成一个立方柱．
游戏开始后，约翰会给贝茜发出P(1≤P≤100000)个指令．指令有两种：
1．移动(M)：将包含X的立方柱移动到包含Y的立方柱上．
2．统计(C):统计名含X的立方柱中，在X下方的方块数目．
写个程序帮贝茜完成游戏．

分析

直接上带权并查集即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=30005;

int dis[N],m,f[N],size[N];

int get_dis(int x)
{
    int ans=0;
    while (f[x]!=x) ans+=dis[x],x=f[x];
    return ans+dis[x];
}

int find(int x)
{
    while (f[x]!=x) x=f[x];
    return x;
}

void merge(int x,int y)
{
    int fx=find(x),fy=find(y);
    if (size[fx]<size[fy]) dis[fx]+=size[fy]-dis[fy],size[fy]+=size[fx],f[fx]=fy;
    else dis[fy]-=dis[fx],dis[fx]+=size[fy],dis[fy]-=size[fy],size[fx]+=size[fy],f[fy]=fx;
}

int main()
{
    for (int i=1;i<=30000;i++) f[i]=i,size[i]=1;
    scanf("%d",&m);
    while (m--)
    {
        char op[2];
        scanf("%s",op);
        if (op[0]=='M')
        {
            int x,y;
            scanf("%d%d",&x,&y);
            merge(x,y);
        }
        else
        {
            int x;
            scanf("%d",&x);
            printf("%d\n",get_dis(x));
        }
    }
    return 0;
}