指南 第三章 例题6 UVALive 3027 Corporative Network（并查集的应用）

题目链接：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;
}