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;
}