【问题描述】
有n个结点(编号为1..n),初始时每个结点的父亲都不存在。你的任务是执行一次I操作和E操作,格式如下:
I u v:把节点u的父亲点设置为v,距离为|u-v|除以1000的余数。输入保证执行指令前u没有父亲节点。
E u:询问u 到根接点的距离(输出距离 mod 1000的结果)。
【输入格式】
输入第一行为测试数据组数T。每组数据第一行为n(5<=n<=20 000)。接下来有不超过20000行,每一行一条指令,以”O”结尾。I指令的个数不小于n。
【输出格式】
对于每条E指令,输出查询结果(输出距离 mod 1000的结果)。
【输入样例】
1
4
E 3
I 3 1
E 3
I 1 2
E 3
I 2 4
E 3
O
【输出样例】
0
2
3
5
分析:
本题连通分量的节点不仅要维护fa信息,还需要维护dist信息。
其中fa[i]表示i节点的直属父亲节点编号,dist[i]表示i节点到它直属父亲节点的距离。在每次findset(i)操作后,将i节点连接到当前连通分量的根节点下面,并且更新dist[i]的值。其实就是比普通的并查集操作多了一个相对距离的概念。
注意:每次查询u节点到根节点距离前,需要findset(u)操作,这样才能让u节点放在其所属连通分量的根节点下面。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=20000+5;
//路径压缩并查集
int fa[maxn];
int dist[maxn];
int findset(int x)
{
if(fa[x]==-1) return x;
int root = findset(fa[x]);
dist[x] += dist[fa[x]];
return fa[x] = root;
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
fa[i]=-1;
dist[i]=0;
}
char str[100];
while(scanf("%s",str)==1 && str[0]!='O')
{
if(str[0]=='I')
{
int u,v;
scanf("%d%d",&u,&v);
fa[u]=v;
dist[u]= abs(u-v)%1000;
}
else if(str[0]=='E')
{
int u;
scanf("%d",&u);
findset(u);
printf("%d\n",dist[u]);
}
}
}
return 0;
}