首页 > 代码库 > HDU 3635 Dragon Balls(带权并查集)

HDU 3635 Dragon Balls(带权并查集)

题目地址: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;
}


HDU 3635 Dragon Balls(带权并查集)