题意:初始时有n个点(编号1到n),每个点都没有父节点,然后依次执行下面两条命令:
思路:并查集+路径压缩维护距离值
#include <cstdio>
#include <queue>
#include <cstring>
#include <iostream>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <map>
#include <string>
#include <set>
#include <ctime>
#include <cmath>
#include <cctype>
using namespace std;
#define maxn 20010
#define LL long long
int cas=1,T;
int pre[maxn];
int dist[maxn];
int n;
void init()
{
for (int i = 0;i<=n;i++)
{
pre[i]=i;
dist[i]=0;
}
}
int Find(int x)
{
if (x!=pre[x])
{
int f = pre[x];
pre[x]=Find(pre[x]);
dist[x]+=dist[f];
return pre[x];
}
else
return x;
// return pre[x];
}
int main()
{
//freopen("in","r",stdin);
scanf("%d",&T);
while (T--)
{
scanf("%d",&n);
init();
char str[10];
while (scanf("%s",str))
{
int a,b;
if (str[0] == 'O')
break;
if (str[0] == 'I')
{
scanf("%d%d",&a,&b);
pre[a]=b;
dist[a]=abs(a-b)%1000;
}
if (str[0] == 'E')
{
scanf("%d",&a);
Find(a);
printf("%d\n",dist[a]);
}
}
}
//printf("time=%.3lf",(double)clock()/CLOCKS_PER_SEC);
return 0;
}