POJ 1962 Corporative Network

思路和正常并查集差不多，只是要注意更新节点到根的距离

 

 

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>

using namespace std;

#define MAXN 20010

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

int find(int a)
{
    
    if(sets[a]!=a)
    {
        int root=find(sets[a]);
        f[a]+=f[sets[a]];                  //记录到跟节点的距离
        sets[a]=root;
        return root;
    }
    else
        return a;
}

bool merge(int a,int b)
{
    int x=find(a);
    int y=find(b);
    if(x==y) return false;
    sets[a]=y; 
    f[a]=abs(a-b)%1000+f[b];        //更新距离值
    return true;
}

int main()
{
    int t;
    char s[20];
    int a,b;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        init();
        while(~scanf("%s",s)&&s[0]!='O')          //避免%c所考虑的getchar()
        {
            if(s[0]=='E')
            {
                scanf("%d",&a);
                find(a); 
                printf("%d\n",f[a]);
            }
            else
            {
                scanf("%d%d",&a,&b);
                merge(a,b);
            }
        }
    }
    return 0;
}