HDU 3635 Dragon Balls(带权并查集)

Posted 2020-09-28 cynchanpin

题目地址：php?

pid=3635">HDU 3635

加权并查集水题。

用num数组维护该城市有多少龙珠，用times数组维护每一个龙珠运输了多少次。num数组在合并的时候维护。times数组因为每一个都不一样。所以要在找根的时候递归来所有维护。

终于，x龙珠所在的城市就是x节点所在的根，x结点所在的跟的num数组值是该城市的龙珠数。times[x]为该龙珠运输了多少次。

代码例如以下：

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <stdlib.h>
#include <math.h>
#include <ctype.h>
#include <queue>
#include <map>
#include <set>
#include <algorithm>

using namespace std;
#define LL long long
int bin[20000], num[20000], times[20000];
int find1(int x)
{
    int y;
    if(bin[x]!=x)
    {
        y=bin[x];
        bin[x]=find1(bin[x]);
        times[x]+=times[y];
    }
    return bin[x];
}
void join(int x, int y)
{
    int f1=find1(x);
    int f2=find1(y);
    if(f1!=f2)
    {
        bin[f1]=f2;
        num[f2]+=num[f1];
        times[f1]++;
    }
}
int main()
{
    int t, nn=0, n, m, i, a, b, f;
    char c;
    scanf("%d",&t);
    while(t--)
    {
        nn++;
        scanf("%d%d",&n,&m);
        for(i=1; i<=n; i++)
        {
            bin[i]=i;
            num[i]=1;
            times[i]=0;
        }
        printf("Case %d:\n",nn);
        while(m--)
        {
            getchar();
            scanf("%c",&c);
            if(c=='T')
            {
                scanf("%d%d",&a,&b);
                join(a,b);
            }
            else
            {
                scanf("%d",&a);
                f=find1(a);
                printf("%d %d %d\n",f,num[f],times[a]);
            }
        }
    }
    return 0;
}