並查集 Corporative Network POJ 1962

輸入E X，表示查找X到根節點的距離；輸入I X Y表示將X連接至Y後（Y作爲父節點）

提示

1)兩點之間的距離要%1000，但是計算到根節點距離時不用%1000

#include <stdio.h>
#include <string.h>
int f[20005];
int num[20005];
int aaa(int a)
{
	return a<0?-a:a;
}
int find(int i)
{
	if(i==f[i])
		return f[i];
	int a=f[i];
	f[i]=find(f[i]);
	num[i]+=num[a];      //遞歸更新num數組的值
	return f[i];
}
int main()
{
	int t,n;
	int i,j;
	char s[3];
	int a,b;
	while(scanf("%d",&t)!=-1)
	{
		while(t--)
		{
			scanf("%d",&n);
			memset(num,0,sizeof(num));
			for(i=0;i<20005;i++)
				f[i]=i;
			while(scanf("%s",s)&&s[0]!='O')
			{
				if(s[0]=='E')
				{
					scanf("%d",&a);
					b=find(a);
					printf("%d\n",num[a]);
				}
				if(s[0]=='I')
				{
					scanf("%d%d",&a,&b);
					f[a]=b;
					num[a]=aaa(a-b)%1000;     //兩個點之間的距離要%1000，但是加和不用
				}
			}
		}
	}
	return 0;
}