UVALive ~ 3027 ~ Corporative Network （带权并查集）

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