poj1962（带权并查集）

题目连接：http://poj.org/problem?id=1962

都是套路。。。

#include<cstdio>
#include<cstring>
const int maxn=20010;
const int mod=1000;
int n;
int f[maxn],r[maxn];

void init()
{
    for(int i=0;i<=n;i++)
    {
        f[i]=i;
        r[i]=0;
    }
}

int gf(int x)
{
    if(x!=f[x])
    {
        int t=f[x];
        f[x]=gf(t);
        r[x]+=r[t];
    }
    return f[x];
}

void uni(int a,int b)
{
    int pa=gf(a);
    int pb=gf(b);
    if(pa!=pb)
    {
        f[pa]=pb;
        int x=(a-b)>=0?a-b:b-a;
        r[pa]=r[b]-r[a]+x%mod;
    }
    return ;
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        int x,y;
        init();
        char s[5];
        while(scanf("%s",s))
        {
            if(s[0]=='O') break;  //看成以0结束，无限tle
            if(s[0]=='E')
            {
                scanf("%d",&x);
                gf(x);
                printf("%d
",r[x]);
            }
            if(s[0]=='I')
            {
                scanf("%d%d",&x,&y);
                uni(x,y);
            }
        }
    }
}