POJ 1962/ ZOJ 2178: Corporative Network - 并查集

简单并查集 

n个结点，初始时啥都不存在，

I u v  操作：把结点u的父节点设为v，距离为|u-v|模除1000；

E u  操作：询问u到根节点的距离。

注意：这里的合并操作指定了父节点和子节点，故不能使用启发式合并

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N=20005;
int p[N],d[N];
int findset(int x)
{
	int root;
	if(x!=p[x]) 
	{
		root=findset(p[x]);
		d[x]+=d[p[x]];
		p[x]=root;
	}
	return p[x];
}

int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		int n,u,v,i;
		char cmd[9];
		scanf("%d",&n);
		for(i=1;i<=n;i++)
		{
			p[i]=i;
			d[i]=0;
		}
		while(scanf("%s",cmd)&&cmd[0]!='O')
		{
			if(cmd[0]=='E'){
				scanf("%d",&u);
				findset(u);
				printf("%d\n",d[u]);
			}
			if(cmd[0]=='I'){
				scanf("%d%d",&u,&v);
				p[u]=v;
				d[u]=abs(u-v)%1000;
			}
		}
	}
	return 0;
}