BZOJ 3376: [Usaco2004 Open]Cube Stacking 方块游戏

本文介绍了一种基于带权并查集的数据结构实现方法,通过维护每个集合中元素的数量和层级来高效处理集合合并及查询操作。文章提供了一个完整的C++代码示例,展示了如何在合并集合时更新元素数量,并在查询时返回指定元素所在集合底部的元素数量。

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

带权并查集

题目传送门

简单的带权并查集,sum[i]表示i这堆有多少个方块,num[i]表示在i的下面有多少方块。找爸爸时更新即可。
注意查询时需要再空找一次爸爸。

代码:

#include<cstdio>
#include<algorithm>
#include<cstring>
#define MAXN 30000
using namespace std;
int fa[MAXN+5],num[MAXN+5],sum[MAXN+5];
int n;
char readc(){
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF;
    else return *l++;
}
inline int _read(){
    int num=0; char ch=readc();
    while ((ch<'0'||ch>'9')&&ch!='M'&&ch!='C') ch=readc();
    if (ch=='M'||ch=='C') return (int)ch;
    while (ch>='0'&&ch<='9'){ num=num*10+ch-48; ch=readc(); }
    return num;    
}
int findfather(int x){
    if (fa[x]==x)
        return x;
    int fat=fa[x];
    fa[x]=findfather(fa[x]);
    num[x]+=num[fat];
    return fa[x];
}
int main(){
    n=_read();
    for (int i=1;i<=MAXN;i++){
        fa[i]=i;
        sum[i]=1;
    }
    while (n--){
        int flag=_read(),x=_read();
        if (flag=='M'){
            int y=_read();
            int fx=findfather(x),fy=findfather(y);
            fa[fx]=fy;
            num[fx]+=sum[fy];
            sum[fy]+=sum[fx];
        }
        if (flag=='C'){
            findfather(x);
            printf("%d\n",num[x]);
        }
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值