hdu 3635 并查集

每个结点除了维护父亲结点编号以外，多维护一个sum值和r值，表示该子树的总结点数（如果该结点是根则sum值有效）以及它到根的距离（即被运输了几次），然后在路径压缩和集合合并的时候顺便维护这两个值即可。

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;

const int N = 10001;
int f[N];
int sum[N];
int r[N];

void init( int n )
{
    for ( int i = 1; i <= n; i++ )
    {
        f[i] = i;
        sum[i] = 1;
        r[i] = 0;
    }
}

int findf( int x )
{
    if ( f[x] == x ) return x;
    int tx = findf(f[x]);
    r[x] = r[x] + r[f[x]];
    f[x] = tx;
    return f[x];
}

void union_set( int x, int y )
{
    x = findf(x), y = findf(y);
    if ( x == y ) return ;
    f[x] = y;
    sum[y] += sum[x];
    r[x]++;
}

int main ()
{
    int t;
    scanf("%d", &t);
    for ( int _case = 1; _case <= t; _case++ )
    {
        printf("Case %d:
", _case);
        int n, m;
        scanf("%d%d", &n, &m);
        init(n);
        while ( m-- )
        {
            char op[2];
            int x, y;
            scanf("%s", op);
            if ( op[0] == 'T' )
            {
                scanf("%d%d", &x, &y);
                union_set( x, y );
            }
            else
            {
                scanf("%d", &x);
                int ans1 = findf(x);
                int ans2 = sum[ans1];
                int ans3 = r[x];
                printf("%d %d %d
", ans1, ans2, ans3);
            }
        }
    }
    return 0;
}