题目链接:http://7xjob4.com1.z0.glb.clouddn.com/8709392d143986024ba5de491366d376
下面写下中文题意:
有n个结点,初始时每个结点的父结点都不存在。你的任务是执行一次I操作和E操作,格式如下:
(1)I u v:把结点u的父结点设为v,距离为|u-v|除以1000的余数。输入保证执行指令前u没有父结点。
(2)E u:询问u到根结点的距离。
【输入格式】
输入第一行为测试数据组数T。每组数据第一行为n(5<=n<=20000)。接下来有不超过20000行,每行一条指令,以“O”结尾。I指令的个数小于n。
【输出格式】
对于每条E指令,输出查询结果。
【分析】
因为题目只查询结点到根结点的距离,所以每棵树除了根结点不能换之外,其他结点的位置可以任意改变,这恰好符合并查集的特点,但是需要记录附加信息。如果记录每个结点到根的距离,那么每次I操作都要更新很多结点的信息,时间复杂度难以保证,因此考虑记下每个结点到父结点的距离为d[i],然后在路径压缩时维护这个d数组。
下面是AC代码:
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=20000 + 10;
int pre[maxn],d[maxn];
int findset(int x)//路径压缩,同时维护d[i];结点i到树根的距离
{
if(x==pre[x])
{
return x;
}
else
{
int root=findset(pre[x]);
d[x]+=d[pre[x]];
return pre[x]=root;
}
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
int n,u,v;
char cmd[9];
scanf("%d",&n);
for(int i=1;i<=n;i++)//初始化,每个结点单独是一棵树
{
pre[i]=i;
d[i]=0;
}
while(scanf("%s",cmd)&&cmd[0]!='O')
{
if(cmd[0]=='E')
{
scanf("%d",&u);
findset(u);
printf("%d\n",d[u]);
}
else
{
scanf("%d%d",&u,&v);
pre[u]=v;
d[u]=abs(u-v)%1000;
}
}
}
return 0;
}