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;
}
©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页