ZCMU--1930: 帽子戏法(C语言)

Description

小A家有很多很多顶帽子初始时帽子都是独立分开的,每顶帽子都有一个编号用于区分.小A会有以下操作之一:
1.将编号为y的帽子所在的帽子堆放在编号为x的帽子所在的帽子堆顶上,如果x,y在同一堆则不做任何动作.
2.小A会向你询问编号为x的帽子上方有多少只帽子.

Input

输入有多组数据:
第一行输入N,M分别为帽子数和操作数(1<=N<=40000,1<=M<=400000)
帽子的编号对应1,2,3...,N.
接下来有M行输入为 T x y 对应操作1. Q x 对应操作2.

Output

对于小A的询问输出位于编号为x的帽子上方的帽子数.

Sample Input

2  2

T  1  2

Q  1

Sample Output

1

解析：一般并查集加上两个额外数组d[ ]和s[ ]来记录编号为 i 的帽子头上有多少帽子，和他下面有多少个帽子(包括他自己)，y移到x上面，相当于x接到y下面，然后更新对应数据即可。

更新数据：x接到y下面，x到顶部的距离其实就是额外加上了y的size大小，y的size额外加上了x的size

#include <stdio.h>
#define N 40005
int f[N],s[N],d[N];//f记录父节点，s记录以他为顶的帽子堆大小，d记录他头上帽子数
int find(int x)
{
	if(f[x]!=x)
	{
		int r=find(f[x]);
		d[x]+=d[f[x]];
		f[x]=r;
	}
	return f[x];
}
int main()
{
	int n,m,i,x,y,x1;
	while(~scanf("%d%d",&n,&m))
	{
		for(i=1;i<=n;i++) f[i]=i,s[i]=1,d[i]=0;//多组，初始化
		while(m--)
		{
			char k[3];
			scanf("%s",k);
			if(k[0]=='T')
			{
				scanf("%d%d",&x,&y);
				x=find(x),y=find(y);
				if(x!=y)
				{
					//不在同一组，更新一下数据
					d[x]=s[y];//x移到y上面相当于y接到x下面
					s[y]+=s[x];
					f[x]=y;
				}
			}else{
				scanf("%d",&x);
				x1=find(x);//可能数据还没更新，通过find来更新再输出
				printf("%d\n",d[x]);
			}
		}	
	}
	return 0;
}