UVALive 3027 Corporative Network （带权并查集）

题意：有 n 个节点，初始时每个节点的父节点都不存在，你的任务是执行一次 I 操作 和 E 操作，含义如下：

I  u  v   :  把节点 u  的父节点设为 v  ，距离为| u - v | 除以 1000 的余数。

E u   : 询问u 到根节点的距离。

解题思路：因为题目只查询节点到根节点的距离，所以每棵树处理根节点不能换之外、其他节点的位置可以随意改变，这恰好符合并查集的特点，但是附加了一点东西、、、、在两点之间有了一个附加的权值（距离）、、所以就是加权并查集了、、题目给的是节点和父节点之间的距离、、但是询问的是节点到根节点的距离、我们可以用 d[i] 表示表示 i 这个节点到 根节点的距离、、在路径压缩的时候维护这个数组、、假设  Y 的父节点是 根节点 ， X 的父节点 是 Y 节点  X 与 Y 之间的 距离 是 L  的d [Y]表示 Y到根节点的距离  ，那么 d[X] = L  +d[Ｙ]，在这里需要递归调用保证Y的父节点一定是根节点，那么d[Y]才是到根节点的距离，如果不是继续寻找 Y 的父节点，使 Y 的父节点的父节点是根节点，如果是那么 d[Y] = d[Y] + d[pre[Y]],（pre[i]代表i 的父节点）如果不是，继续递归调用，一直到是为止~~

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <cctype>
#include <algorithm>
using namespace std;
const int MAXN = 2e4 + 3;
const int MOD = 1000;
int pre[MAXN]; 
int d[MAXN];//存放的是节点到根节点的距离

int FindWeight(int x) //查找压缩的同时维护数组d[i].
{
    if(pre[x] != x){                    
        int root = FindWeight(pre[x]);    //递归调用，保证 X 的父节点 pre[X]的父节点是根节点。
        d[x] += d[pre[x]];
        return pre[x] = root;
    }
    else
        return x;
}

int main()
{
    //freopen("in.cpp","r",stdin);
    //fropen("out.txt","w",stdout);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int N;
        scanf("%d",&N);
        for(int i = 1; i <= N; i++)     //初始化
        {
            pre[i] = i;
            d[i] = 0;
        }
        char s[3]={0};
        while(~scanf("%s",s) && (s[0]!='O') )
        {
            if(s[0] == 'E')
            {
                int u;
                scanf("%d",&u);
                FindWeight(u);
                printf("%d\n",d[u]);

            }
            if(s[0] == 'I')
            {
               // cout <<endl;
                int u,v;                    //v是u的父节点
                scanf("%d%d",&u,&v);
                pre[u] = v;
                d[u] = abs(u-v)%MOD;
            }
        }
    }
    return 0;
}