题意:有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 */