LA3027 合作网络

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<stdlib.h>
#include<math.h>
#include<queue>
#include<vector>
#include<map>
#define MAXN 20000+10
//#define MAXM 20000+5
#define oo 9556531
#define eps 0.000001
#define PI acos(-1.0)
#define REP1(i,n) for(int i=0;i<(n);i++)
#define REP2(i,n) for(int i=1;i<=(n);i++)
using namespace std;

int p[MAXN];
int d[MAXN];
int n,t;

void init()
{
    REP2(i,n)
    {
        p[i]=i;
        d[i]=0;
    }
}

int findx(int x)
{
//    return p[x]==x?0:d[x]+getans(p[x]);
      if (p[x]!=x) //d[x]任然为0，不需要更新
      {
          int px=findx(p[x]);//压缩栈，保证前面的根节点已经更新过
          d[x]=d[x]+d[p[x]];//保证d[p[x]]已经是p[x]到根节点的距离
//          cout<<"px="<<px<<","<<"d[x]="<<d[x]<<endl;
          return p[x]=px;//在所有的d更新完毕后，把x连接到根节点px上
      }else return x;
}

int main()
{
    cin>>t;
    for(;t;t--)
    {
        scanf("%d
",&n);
        init();
        char Q[20];
        while(cin>>Q)
        {
//            cout<<"Q="<<Q<<endl;
            if (Q[0]=='O') break;
            if (Q[0]=='E')//多次询问E，动态更新
            {
                int x;cin>>x;
                int px=findx(x);
                cout<<d[x]<<endl;
            }
            if (Q[0]=='I')
            {
                int i,j;
                cin>>i>>j;
                d[i]=abs(i-j)%1000;//d[i]：i到父节点的距离,注意取模
                p[i]=j;//i的父节点是j
            }
        }
    }
    return 0;
}