算法竞赛训练指南192页,带权并查集
本题要点:
1、命令处理:
查询命令,E u, 先查询 u 的父节点, get(u), 然后 输出 u到根节点的距离 d[u].
查询的时候,使用了路径压缩。同时累加 d[x] += d[fa[x]]
2、I u v ,把 u的父节点设为v, 题目说明了当前的u点是没有父节点的。
fa[x] = y, d[x] = abs(x - y) % 1000;
#include <cstdio>
#include <cstring>
#include <iostream>
#include <cstdlib>
using namespace std;
const int MaxN = 20010;
int fa[MaxN], d[MaxN]; // d[i] 表示距离根节点的距离
int T, n;
char ch[2] = {0};
int get(int x)
{
if(fa[x] == x)
return x;
int root = get(fa[x]);
d[x] += d[fa[x]];
return fa[x] = root;
}
int main()
{
int x, y;
scanf("%d", &T);
while(T--)
{
scanf("%d", &n);
for(int i = 0; i <= n; ++i)
{
fa[i] = i, d[i] = 0;
}
while(scanf("%s", ch))
{
if(ch[0] == 'O')
break;
if(ch[0] == 'E') //查询
{
scanf("%d", &x);
get(x);
printf("%d\n", d[x]);
}else{ //移动
scanf("%d%d", &x, &y);
fa[x] = y, d[x] = abs(x - y) % 1000;
}
}
}
return 0;
}
/*
1
4
E 3
I 3 1
E 3
I 1 2
E 3
I 2 4
E 3
O
*/
/*
0
2
3
5
*/