LA3027 - Corporative Network（结点到根的距离）

题目链接

简介：
有n个结点，初始时每个结点都是独立的，有两种操作：
I u v：把u的父节点设为v，距离为|u-v|%1000
E u：询问u到根节点的距离

分析：
题目只查询结点到根节点的距离，所以整棵树除了根节点不能变换之外，其他结点的位置可以任意变换

这恰好符合并查集的特点

为了快速回答询问，我们需要记录每个节点到父亲的距离d
在并查集的路径压缩的同时维护d

注意：

由于本题的合并操作指定了父节点和儿子结点，所以不能用按秩合并

tip

fa和d的初始化不要忘了

//这里写代码片
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>

using namespace std;

const int N=20003;
int n,fa[N],d[N];

int find(int x)
{
    if (fa[x]!=x)
    {
        int root=find(fa[x]);                            
        d[x]+=d[fa[x]];                            //维护d，此时的fa还是没有压缩的 
        fa[x]=root;
    }
    return fa[x];
}

int main()
{
    int T;
    scanf("%d",&T);
    while (T--)
    {
        scanf("%d",&n);
        for (int i=1;i<=n;i++) fa[i]=i,d[i]=0;      //初始化
        char opt[2];
        while (scanf("%s",&opt)!=EOF&&opt[0]!='O')
        {
            int u,w;
            if (opt[0]=='I'){
                scanf("%d%d",&u,&w);
                fa[u]=w;                           //直接连接 
                d[u]=abs(u-w)%1000;                //不要忘了计算距离 
            }
            else{
                scanf("%d",&u);
                find(u);                           //在询问的时候路径压缩一次就可以了 
                printf("%d\n",d[u]);
            }
        } 
    }
    return 0;
}