[US Open 2004][luogu2342] 叠积木 [带权并查集]

题面

洛谷传送门

思路

害

学了4年多OI，第一次知道还有带权并查集这个东西

wtcl

这个玩意儿的原理和详细实现，可以参考这个博客：带权并查集传送门

这道题，就是在带权并查集的基础上，加个维护每个集合的大小。

并查集往每堆积木的底部那个上面合并（也就是根是最底下的积木），合并的时候把儿子的value设定成父亲的size即可

Code

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cassert>
#define ll long long
using namespace std;
inline int read(){
	int re=0,flag=1;char ch=getchar();
	while(ch>'9'||ch<'0'){
		if(ch=='-') flag=-1;
		ch=getchar();
	}
	while(ch>='0'&&ch<='9') re=(re<<1)+(re<<3)+ch-'0',ch=getchar();
	return re*flag;
}
int n=30000,m,f[100010],siz[100010],val[100010];
inline int find(int x){//带权并查集的路径压缩+权值维护
	if(x==f[x]) return x;
	int tmp=f[x];
	f[x]=find(f[x]);
	val[x]+=val[tmp];
	return f[x];
}
int main(){
	m=read();int i;
	char s[10];int t1,t2,x,y;
	for(i=1;i<=n;i++) f[i]=i,siz[i]=1,val[i]=0;
	while(m--){
		scanf("%s",s);
		if(s[0]=='M'){
			t1=read();t2=read();
			x=find(t1);
			y=find(t2);
			f[x]=y;//合并到底部
			val[x]=siz[y];//权值和集合大小的关系
			siz[y]+=siz[x];
		}
		else{
			t1=read();
			x=find(t1);
			printf("%d\n",val[t1]);
		}
	}
}