【例题】【并查集(带权)&题干有诈】NKOJ 3764 树上间距

NKOJ 3764 树上间距

时间限制 : - MS 空间限制 : 65536 KB

评测说明 : 时限1000ms

问题描述

有n个节点，初始时每个节点的父亲节点都不存在。你的任务是执行下列两种操作：

1 x y 把节点x的父亲设为y,距离为|x-y| mod 1000 输入保证执行指令前x没有父亲节点

2 x 询问x到它所在这棵树的根节点的距离

输入输入

第一行，两个整数n(5<=n<=50000)

接下来若干行(行数<=100000)，每行代表一个操作

输出格式

对于每个2号操作，输出一行，表示计算结果

样例输入

4

2 3

1 3 1

2 3

1 1 2

2 3

1 2 4

2 3

样例输出

0

2

3

5

来源 改编自la3027

思路：

使用并查集存储点的父子关系，一个集合中元素的dis值表示该点到根的距离。

当查询x所在集合根节点时，更新dis[x]。

(有诈：)注意权值只有第一次添加时mod1000，更新时不mod

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstdlib>
using namespace std;
const int need=50004;

int be[need],dis[need];
int d;

int getbe(int x)
{
if(x==be[x]) return x;
else
{
int root=getbe(be[x]);
dis[x]+=dis[be[x]];
return be[x]=root;
}
}

int main()
{
//freopen("a.txt","r",stdin);
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++) be[i]=i;
int k,a,b;
while(scanf("%d",&k)!=EOF)
{
if(k==1)
{
scanf("%d%d",&a,&b);
be[a]=b;
dis[a]=abs(a-b)%1000;
}
else if(k==2)
{
scanf("%d",&a);
getbe(a);//注意要使用dis时，先更新根节点
printf("%d\n",dis[a]);
}
}
}