Uva 3027 Corporative Network

本题只查询结点到根节点的距离，所以每颗树除了根节点不能换之外，其他结点的位置可以任意改变，这恰好符合并查集的特点，但是需要记录附加信息。如果记录每个结点到根节点的距离，那么每次I操作都要更新很多结点的信息，时间复杂度难以保证，因此考虑记下每个结点到父结点的距离为d[i]，然后在路径压缩是维护这个数组d。

#include <queue>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int MAX = 20010;

int d[MAX],father[MAX],n;

void Init(){
	for(int i=1; i<=n; i++){
		d[i] = 0;
		father[i] = i;
	}
}

int Find(int x){
	if(x != father[x]){
		int root = Find(father[x]);//先记录以下递归的结果
		d[x] += d[father[x]];
		return father[x] = root;//继续递归
	}
	else return x;
}

int main(){
	int cas;
	scanf("%d",&cas);
	while(cas--){
		scanf("%d",&n);
		Init();
		char s[10];
		while(scanf("%s",s) == 1){
			int u,v;
			if(s[0] == 'O') break;
			if(s[0] == 'E'){
				scanf("%d",&u);
				int tmp = Find(u);
				cout << d[u] << endl;
			}		
			else{
				scanf("%d%d",&u,&v);
				father[u] = v;
				d[u] = abs(u-v)%1000;
			}
		}
	}
	return 0;
}