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

最新推荐文章于 2022-11-17 09:19:22 发布

forezxl

最新推荐文章于 2022-11-17 09:19:22 发布

阅读量389

点赞数 1

CC 4.0 BY-SA版权

分类专栏： BZOJ 数据结构---并查集蒟蒻zxl的Blog专栏文章标签：带权并查集并查集 BZOJ

本文链接：https://blog.youkuaiyun.com/a1799342217/article/details/78013197

蒟蒻zxl的Blog专栏同时被 3 个专栏收录

460 篇文章

订阅专栏

BZOJ

281 篇文章

订阅专栏

数据结构---并查集

14 篇文章

订阅专栏

本文介绍了一种基于带权并查集的数据结构实现方法，通过维护每个集合中元素的数量和层级来高效处理集合合并及查询操作。文章提供了一个完整的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;
}