连通块中点的数量

连通块中点的数量

输入样例：

5 5
C 1 2
Q1 1 2
Q2 1
C 2 5
Q2 5

输出样例：

Yes
2
3

模板：

//连通块中点的数量
#include<iostream>
using namespace std;
const int N = 100010;
int p[N],size[N];
int n,m; 
int find(int x){
	if(p[x]!=x) p[x]=find(p[x]);
	return p[x];
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		p[i]=i;
		size[i]=1;
	}
	while(m--){
		char op[5];
		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)) printf("Yes\n");
			else printf("No\n");
		}else{
			scanf("%d",&a);
			printf("%d\n",size[find(a)]);
		}
	}
	return 0;
} 

---