《信息学奥赛一本通》1389：亲戚

1389：亲戚

时间限制: 1000 ms 内存限制: 65536 KB
提交数: 4294 通过数: 1358
【题目描述】
若某个家族人员过于庞大，要判断两个是否是亲戚，确实还很不容易，现在给出某个亲戚关系图，求任意给出的某个人所在家族的人数。

规定：x和y是亲戚，y和z是亲戚，那么x和z也是亲戚。如果x,y是亲戚，那么x的亲戚都是y的亲戚，y的亲戚也都是x的亲戚。

【输入】
第一行：三个整数n，（n≤100,000,m≤200,000），分别表示有n个人，m个信息。

以下m行：信息包含两种形式：

M a b：表示a和b具有亲戚关系。

Q a：要求输出a所在家族的人数。

【输出】
要求输出a所在家族的人数。

【输入样例】
5 10
M 3 2
Q 4
M 1 2
Q 4
M 3 2
Q 1
M 3 1
Q 5
M 4 2
Q 4
【输出样例】
1
1
3
1
4

解析：
用一个num数组初始化为1，表示该家族目前只有一个人
并查集合并祖先的时候顺带把家族人数加在一起就好啦~

当然不要忘了路径优化，不然…

#include<bits/stdc++.h>
#define N 100001
using namespace std;

int n,m,a,b;
int dad[N],num[N];			//num存储家族人数 
char s[10];

int FIND(int x)
{
	if(dad[x]==x)
		return x;	
	return dad[x]=FIND(dad[x]);//路径压缩不能忘，不然只有50分了

}

void UNION(int fa,int so)
{
	fa=FIND(fa);
	so=FIND(so);
	if(fa!=so)
	{
		dad[so]=fa;
		num[fa]+=num[so];		
	}
}

int main()
{
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		dad[i]=i;
		num[i]=1;
	}
	for(int i=1;i<=m;i++)
	{
		scanf("%s",s);
		if(s[0]=='M')
		{
			scanf("%d %d",&a,&b);
			UNION(a,b);
		}
		if(s[0]=='Q')
		{
			scanf("%d",&a);
			printf("%d\n",num[FIND(a)]);
		}
	}
	return 0;
}