ACWing837 模板题 连通块中点的数量 （并查集）

题目链接

并查集操作：合并 查询两元素是否在同一集合 统计联通块大小size

 

在进行并查集操作的同时维护联通块中节点的数量

初始化 size[i]=1

合并时 在当前新的根节点上 加上这棵树的节点个数 size[find(b)]+=size[find(a)]

 

模板代码如下

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=100005;
int size[N],p[N];//只保证根节点的size有意义 
int find(int x){
	if(p[x]!=x) p[x]=find(p[x]);
	return p[x];
}
int main(){
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;++i){
		p[i]=i;
		size[i]=1;
	}
	while(m--){
		char op[3];
		int a,b;
		scanf("%s",&op);
		if(op[0]=='C'){
			scanf("%d%d",&a,&b);
			if(find(a)==find(b))continue;//相等特判 不计数 
			
			size[find(b)]+=size[find(a)];//这两句话的
			p[find(a)]=find(b);//先后顺序要注意
			
		}else if(op[1]=='1'){
			scanf("%d%d",&a,&b);
			if(find(a)==find(b)) puts("Yes");
			else puts("No");
		}else{
			scanf("%d",&a);
			printf("%d\n",size[find(a)]);//
		}
	}
}