合作网络（Corporative Network,LA 3027 )

题目链接

https://vjudge.net/problem/UVALive-3027

题意

有n个结点，初始时每个结点的父节点都不存在。你的任务是执行一次I操作和E操作，格式如下
I u v:把结点u的父节点设为v，距离为|u-v|除以1000的余数。输入保证执行指令前u没有父节点。
E u: 询问u到根结点的距离

分析

模拟一遍样例，注意到每次I操作都会把u的父节点改变成v的父节点（初始时每个结点的父节点是自身）。这和并查集的操作是一样的。在每次寻找u的根的过程中，进行路径压缩后，u的父节点都会变成根。

同时，还需要维护一个权值dis[i]，表示结点i到根的距离。同样是在路径压缩时进行维护，dis[u]=dis[u]+dis[par[u]].

代码

#include <bits/stdc++.h>
using namespace std;
const int maxn=2e4+100;
int dis[maxn],par[maxn];

int find(int x)
{
    if(par[x]==x) return x;
    int rt=find(par[x]);
    dis[x]+=dis[par[x]];
    return par[x]=rt;
}
void init(int n)
{
    for(int i=0;i<=n;i++) par[i]=i;
    memset(dis,0,sizeof(dis));
}
int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        int n;
        cin>>n;
        init(n);
        char op;
        int u,v;
        while(cin>>op)
        {
            if(op=='O') break;
            cin>>u;
            if(op=='I')
            {
                cin>>v;
                par[u]=v;
                dis[u]=abs(u-v)%1000;
            }
            else
            {
                find(u);
                cout<<dis[u]<<endl;
            }
        }
    }
    return 0;
}