一、题意解析
\(n\)个节点,若干次询问,\(I\) \(x\) \(y\)表示从\(x\)连一条边到\(y\),权值为\(|x-y|%1000\);\(E\) \(x\)表示询问\(x\)到\(x\)所指向的终点的距离。
二、实现代码
#include <bits/stdc++.h>
using namespace std;
const int N = 20010;
int fa[N];
int d[N]; // d[i] 表示距离根节点的距离
int n;
//带权并查集模板
int find(int x) {
//点权的理解:每个点到根的距离
if (fa[x] == x) return x; //自己是老大,返回自己
int root = find(fa[x]); //通过自己父亲找老大
d[x] += d[fa[x]]; //点权在路径压缩过程中,需要加上父亲的点权
return fa[x] = root; //路径压缩
}
int main() {
int T;
cin >> T;
while (T--) {
cin >> n;
// 带权并查集初始化
// 1、自已是自己的祖先
// 2、自己距离自己的距离是0,因为在全局变量区创建,无需专门初始化
for (int i = 0; i <= n; i++) fa[i] = i;
char op;
while (cin >> op) {
int x, y;
if (op == 'O') break;
if (op == 'E') { //查询
cin >> x;
find(x); //实现路径压缩,合并了x到根的权值
printf("%d\n", d[x]); // x到根的权值,记录在d[x]中
} else {
//把x的父节点设为y,题目说明了当前的x点是没有父节点的。
cin >> x >> y;
//带权并查集中,权是记录在点上的
fa[x] = y, d[x] = abs(x - y) % 1000;
}
}
}
return 0;
}