poj 1962 Corporative Network（带权并查集）

题意：

在n个站点间建电线；两种操作：

I a b表示以a为中心站点建线；

E a表示查询以a站点为中心，相连的电线总长度；

思路：

带权并查集；中心站点就是父亲，电线长度为权值；

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int fa[500010],w[500010];
int t,n,m;
char ch[5];
void init()
{
    for(int i=0;i<=n;i++)
    {
        fa[i]=i;w[i]=0;
    }
}
int find(int x)
{
    if(x==fa[x]) return x;
    int temp=fa[x];
    fa[x]=find(fa[x]); //路径压缩
    w[x]+=w[temp];  //权值相加
    return fa[x];
}
void combine(int a,int b)
{
    fa[a]=b;
    w[a]=abs(a-b)%1000; //赋权值
}
int main()
{
    int i,j,k,a,b;
    while(scanf("%d",&t)!=EOF)
    {
        while(t--)
        {
            scanf("%d",&n);
            init();
            while(scanf("%s",ch))
            {
                if(ch[0]==‘O‘) break;
                else if(ch[0]==‘E‘){
                    scanf("%d",&a);
                    find(a);
                    printf("%d\n",w[a]);
                }
                else{
                    scanf("%d%d",&a,&b);
                    combine(a,b);
                }
            }
        }
    }
    return 0;
}

版权声明：本文为博主原创文章，未经博主允许不得转载。