题目链接:http://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=4075
I u v:把u的父亲结点设为v,距离为|u-v|%1000.输入保证执行指令前u没有父节点。
E u:询问u到根结点的距离。
#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;
}