uvalive 3027 Corporative Network

https://vjudge.net/problem/UVALive-3027

题意：

有n个点，两种操作：

1.E u : 计算u到根节点的距离；

2.I u v : 把v变成u的父亲，并且把它们之间的距离赋值为 |v-u| % 1000。

现在给出若干个这样的操作，对于每个E输出查询结果。

思路：

裸的带权并查集，每次路径压缩都要有d[u] += d[par[u]]；然后每次合并的时候是有方向的，因为指定了父亲和儿子。

代码：

#include <stdio.h>
#include <string.h>

int d[20005];
int par[20005];

int mabs(int x)
{
    if (x >= 0) return x;
    else return -x;
}

void init(int n)
{
    for (int i = 0;i <= n;i++) par[i] = i;

    memset(d,0,sizeof(d));
}

int fin(int x)
{
    if (x == par[x])
    {
        return x;
    }
    else
    {
        int p = par[x];

        par[x] = fin(p);

        d[x] += d[p];

        return par[x];
    }
}

int main()
{
    int t;

    scanf("%d",&t);

    while (t--)
    {
        int n;

        scanf("%d",&n);

        init(n);

        for (;;)
        {
            char s[20];

            scanf("%s",s);

            if (s[0] == 'O') break;

            if (s[0] == 'E')
            {
                int x;

                scanf("%d",&x);

                fin(x);

                printf("%d
",d[x]);
            }
            else
            {
                int x,y;

                scanf("%d%d",&x,&y);

                d[x] = mabs(x - y)  % 1000;

                par[x] = y;
            }
        }
    }

    return 0;
}