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;
}