【例题】【并查集(带权)&题干有诈】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]);
        }
    }
}