POJ 1962-Corporative Network （并查集）

题目有2种操作， 一种是查询，一种是设置。

设置为将u的父亲设置为v，然后他们之间的距离为|u-v|%1000

查询为该点到根点的距离

用并查集做，做的时候注意维护即可，注意取余操作。

代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
const int N = 200010;
int d[N],F[N];
int init(int x)
{
    for(int i =0;i<=x;i++)
    {
        F[i] = i;
        d[i]=0;
    }
}
int findset(int x)
{
    if(F[x]!=x)
    {
        int root = findset(F[x]);
        d[x] = d[x]+d[F[x]];
        return F[x] = root;
    }
    else return x;
}
int main()
{
    int T,n;
    char s[10];
    scanf("%d",&T);
    while(T--)
    {
        int n,u,v;
        scanf("%d",&n);
        init(n);
        while(scanf("%s",s) && s[0]!='O')
        {
              if(s[0] == 'E')
              {
                scanf("%d",&u);
                findset(u);
                printf("%d\n",d[u]);
              }
              else
              {
                  scanf("%d%d",&u,&v);
                  F[u] = v;
                  d[u] = abs(u-v)%1000;
              }
        }
    }
    return 0;
}