hdu 3635 并查集

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

 1 #include <iostream>
 2 #include <cstring>
 3 #include <cstdio>
 4 using namespace std;
 5
 6 const int N = 10001;
 7 int f[N];
 8 int sum[N];
 9 int r[N];
10
11 void init( int n )
12 {
13     for ( int i = 1; i <= n; i++ )
14     {
15         f[i] = i;
16         sum[i] = 1;
17         r[i] = 0;
18     }
19 }
20
21 int findf( int x )
22 {
23     if ( f[x] == x ) return x;
24     int tx = findf(f[x]);
25     r[x] = r[x] + r[f[x]];
26     f[x] = tx;
27     return f[x];
28 }
29
30 void union_set( int x, int y )
31 {
32     x = findf(x), y = findf(y);
33     if ( x == y ) return ;
34     f[x] = y;
35     sum[y] += sum[x];
36     r[x]++;
37 }
38
39 int main ()
40 {
41     int t;
42     scanf("%d", &t);
43     for ( int _case = 1; _case <= t; _case++ )
44     {
45         printf("Case %d:\n", _case);
46         int n, m;
47         scanf("%d%d", &n, &m);
48         init(n);
49         while ( m-- )
50         {
51             char op[2];
52             int x, y;
53             scanf("%s", op);
54             if ( op[0] == ‘T‘ )
55             {
56                 scanf("%d%d", &x, &y);
57                 union_set( x, y );
58             }
59             else
60             {
61                 scanf("%d", &x);
62                 int ans1 = findf(x);
63                 int ans2 = sum[ans1];
64                 int ans3 = r[x];
65                 printf("%d %d %d\n", ans1, ans2, ans3);
66             }
67         }
68     }
69     return 0;
70 }