uvalive 3027 Corporative Network（种类并查集）

题意：

题目就是要你输出各个点与根节点的距离。有两种指令

I a b，的意思：将b作为a的根节点，a和b之间的距离为|a-b|%1000。

E a， 的意思：求出a与根节点的距离。

思路：

并查集的变形，在压缩路径的时候顺便在更新一下与根节点的距离。

ps：记录与根节点的距离要再开一个数组，我在更新距离的时候还是写错了，太菜。

#include<iostream>
#include<cstdio>
#include<string>
#include<cmath>
using namespace std;
const int mx = 21000;
int fa[mx],d[mx];
int n;
void init(){
	for(int i = 1; i<mx ;i++){
		d[i] = 0;
		fa[i] = i;
	}
}
int find(int x){
	if(x != fa[x]){
		int te = fa[x];//一定要在压缩路径之前记录自己原先的父节点，方便更新
		fa[x] = find (fa[x]);
		d[x] += d[te];
	} 
	return fa[x];
}


 

int main(){
	int T;
	scanf("%d",&T); 
	while(T--){
		scanf("%d",&n);
		init();
		char s;
		int a, b;
		while(cin>>s && s != 'O'){
			if(s == 'E') {
				scanf("%d",&a);
				find(a);
			printf("%d\n",d[a]);
			}
			else {
				scanf("%d%d",&a,&b);
				fa[a] = b;
				d[a] = abs(a - b) % 1000;
			}
		}
	}	
}