题意:
一开始有n个点,每个点都没有父节点。接下来不断执行I操作和E操作,直到执行O操作时结束;
I操作:I u v,u的父节点为v,路线长度为|u-v|%1000,保证u不存在起始父节点。
E操作:E u,查询u到其根节点的距离,输出。
题解:
并查集+路径压缩。
由于I操作时保证了u开始没有父节点,所以构成的是树;之后只是求点到跟的距离,所以只要用并查集即可,之后用路径压缩减少计算路径的时间。f[x]一开始表示的是x的父节点,之后一次并查集查询后f[x]表示的是x的根节点;在查询过程中c[x]表示x到f[x]的距离,而在查找的时候还会维护所有路径上的点到根节点的距离,减少了下次查询的时间。
代码:
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
#include <queue>
#include <stack>
using namespace std;
const int maxn=2e4+10;
int f[maxn],c[maxn];
int findset(int x)
{
if(f[x]!=x)
{
int t=findset(f[x]);
c[x]+=c[f[x]];
f[x]=t;
}
return f[x];
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int i,j,k,n,u,v;
char s[2];
scanf("%d",&n);
for(i=0;i<=n;i++)f[i]=i,c[i]=0;
while(scanf("%s",s)&&s[0]!='O')
{
if(s[0]=='I')
{
scanf("%d%d",&u,&v);
f[u]=v;
c[u]=abs(u-v)%1000;
}
else
{
scanf("%d",&u);
findset(u);
printf("%d\n",c[u]);
}
}
}
return 0;
}