思路和正常并查集差不多,只是要注意更新节点到根的距离
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
#define MAXN 20010
int f[MAXN];
int sets[MAXN];
int n;
void init()
{
for(int i=0;i<=n;i++)
sets[i]=i,f[i]=0;
}
int find(int a)
{
if(sets[a]!=a)
{
int root=find(sets[a]);
f[a]+=f[sets[a]]; //记录到跟节点的距离
sets[a]=root;
return root;
}
else
return a;
}
bool merge(int a,int b)
{
int x=find(a);
int y=find(b);
if(x==y) return false;
sets[a]=y;
f[a]=abs(a-b)%1000+f[b]; //更新距离值
return true;
}
int main()
{
int t;
char s[20];
int a,b;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
init();
while(~scanf("%s",s)&&s[0]!='O') //避免%c所考虑的getchar()
{
if(s[0]=='E')
{
scanf("%d",&a);
find(a);
printf("%d\n",f[a]);
}
else
{
scanf("%d%d",&a,&b);
merge(a,b);
}
}
}
return 0;
}