题意:有n个结点,初始时每个结点的父节点都不存在。你的任务是执行一次I操作和E操作,格式如下。
□I u v:把结点u的父节点设为v,距离为|u-v|%1000的余数。输入保证执行指令前u没有父节点。
□E u:询问u到根节点的距离
【输入个数】
输入第一行为测试数据组数T。每组数据第一行为n(5≤n≤20000)。接线来有不超过20000行,每行一条指令,以‘O’结尾。I指令的个数小于n。
【输出格式】
对于每条E指令,输出查询结果
思路:之前有学过带权并查集,所以。。。注意每次E指令的时候应该在查询一次在输出,因为可能此时还没路径压缩,W[x]表示的还是到父节点的权值,而不是根节点。
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 2e4 + 5;
int n, f[MAXN], w[MAXN];
void init()
{
for (int i = 0; i <= n; i++) f[i] = i, w[i] = 0;
}
int Find(int x)
{
if (f[x] == x) return x;
int t = f[x];
f[x] = Find(f[x]);
w[x] = w[x] + w[t];
return f[x];
}
int main()
{
int T; scanf("%d", &T);
while (T--)
{
scanf("%d", &n);
init();
char op[5];
while (~scanf("%s", &op) && op[0] != 'O')
{
int u, v;
if (op[0] == 'I')
{
scanf("%d%d", &u, &v);
f[u] = v; w[u] = abs(u-v)%1000;
}
if (op[0] == 'E')
{
scanf("%d", &u);
int x = Find(u);
printf("%d\n", w[u]);
}
}
}
return 0;
}
/*
1
4
E 3
I 3 1
E 3
I 1 2
E 3
I 2 4
E 3
O
*/