LA 3027 合作网络（并查集）(训练指南)

#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn =20000+10;
int pa[maxn],d[maxn];
int findset(int x)///路径压缩，同时维护d[i]:结点i到树根的距离
{
if(pa[x]!=x)
{
int root=findset(pa[x]);
d[x]+=d[pa[x]];
return pa[x]=root;
}
else return x;
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int n,u,v;
scanf("%d",&n);
for(int i=1;i<=n;i++)///初始化
pa[i]=i,d[i]=0;
char m[9];
while(scanf("%s",m)&&m[0]!='O')
{
if(m[0]=='E')
{
scanf("%d",&u);
findset(u);
printf("%d\n",d[u]);
}
else
{
scanf("%d%d",&u,&v);
pa[u]=v;
d[u]=abs(u-v)%1000;
}
}
}
return 0;
}