题意:
题目就是要你输出各个点与根节点的距离。有两种指令
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;}}}
}