uva1329 Corporative Network(并查集：路径压缩)

题意：初始时有n个点(编号1到n)，每个点都没有父节点，然后依次执行下面两条命令：

        I u v ：把节点u的父亲设置为v，并且设u节点到v的距离为|u-v|%1000，输入保证执行I命令的时候，u没有父节点。

        E u：查询u节点到根节点的距离。

        输入：首先是T，表示实例个数。每个实例第一行是n（5<=n<=20000），接下有不超过200000行，每行一个指令，以单独的O字母行表示结束。

        输出：对于每条E指令，输出查询结果。

思路：并查集+路径压缩维护距离值

#include <cstdio>
#include <queue>
#include <cstring>
#include <iostream>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <map>
#include <string>
#include <set>
#include <ctime>
#include <cmath>
#include <cctype>
using namespace std;
#define maxn 20010
#define LL long long
int cas=1,T;
int pre[maxn];
int dist[maxn];
int n;
void init()
{
	for (int i = 0;i<=n;i++)
	{
		pre[i]=i;
		dist[i]=0;
	}
}
int Find(int x)
{
	if (x!=pre[x])
	{
        int f = pre[x];
		pre[x]=Find(pre[x]);
		dist[x]+=dist[f];
		return pre[x];
	}
	else
		return x;
//	return pre[x];
}
int main()
{
	//freopen("in","r",stdin);
	scanf("%d",&T);
	while (T--)
	{
		scanf("%d",&n);
		init();
		char str[10];
		while (scanf("%s",str))
		{
			int a,b;
			if (str[0] == 'O')
				break;
			if (str[0] == 'I')
			{
                scanf("%d%d",&a,&b);
				pre[a]=b;
                dist[a]=abs(a-b)%1000;
			}
			if (str[0] == 'E')
			{
                scanf("%d",&a);
				Find(a);
				printf("%d\n",dist[a]);
			}
		}
	}
	//printf("time=%.3lf",(double)clock()/CLOCKS_PER_SEC);
	return 0;
}