HDU 3635 Dragon Balls

涉及到当前所在位置，当前位置总数，和当前移动次数。并查集也就是一棵自底向上更新的树，Lazy操作是递归的Find(Pa[u])之后再更新与父节点的信息。当然像查大小这类只会在根节点进行的操作就没必要PushDown了。

#include <algorithm>
#include <iostream>
#include <fstream>
#include <sstream>
#include <iomanip>

#include <map>
#include <set>
#include <list>
#include <stack>
#include <queue>
#include <deque>
#include <vector>
#include <string>
#include <bitset>
#include <memory>
#include <complex>
#include <numeric>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <time.h>
#include <ctype.h>
#include <locale.h>

using namespace std;

#pragma pack(4)

const double  eps = 1e-8;
const double   pi = acos(-1.0);
const int     inf = 0x7f7f7f7f;

#define loop(a,n)                            \
    for(int i=0;n>i;i++)                     \
        cout<<a[i]<<(i!=n-1?' ':'\n')
#define loop2(a,n,m)                         \
    for(int i=0;n>i;i++)                     \
        for(int j=0;m>j;j++)                 \
            cout<<a[i][j]<<(j!=m-1?' ':'\n')

#define   at(a,i) ((a)&(1<<(i)))
#define   nt(a,i) ((a)^(1<<(i)))
#define set1(a,i) ((a)|(1<<(i)))
#define set0(a,i) ((a)&(~(1<<(i))))

#define cmp(a,b) (fabs((a)-(b))<eps?0:(((a)-(b))>eps?+1:-1))

#define lmax(a,b) ((a)>(b)?(a):(b))
#define lmin(a,b) ((a)<(b)?(a):(b))
#define fmax(a,b) (cmp(a,b)>0?(a):(b))
#define fmin(a,b) (cmp(a,b)<0?(a):(b))

const int MAXV = 10002;

class UFS
{
private:
    int Pa[MAXV],Sz[MAXV],Mv[MAXV];
public:
    void clear(int l,int r)
    {
        for(int i=l;i<=r;i++)
        {
            Pa[i]=i;
            Sz[i]=1;
            Mv[i]=0;
        }
    }
    int Find(int u)
    {
        if(Pa[u]!=u)
        {
            int root=Find(Pa[u]);
            Mv[u]+=Mv[Pa[u]];
            return Pa[u]=root;
        }
        else return u;
    }
    void Union(int u,int v)
    {
        int uu=Find(u);
        int vv=Find(v);
        Pa[uu]=vv;
        Sz[vv]+=Sz[uu];
        Mv[uu]++;
    }
    int query_at(int u)
    {
        return Find(u);
    }
    int query_cnt(int u)
    {
        return Sz[Find(u)];
    }
    int query_move(int u)
    {
        return Find(u),Mv[u];
    }
}ufs;

char o;
int tcase,times=1,n,m,u,v;

int main()
{
    #ifndef ONLINE_JUDGE
    freopen("Dragon Balls.txt","r",stdin);
    #else
    #endif

    scanf("%d",&tcase);while(tcase--)
    {
        scanf("%d %d",&n,&m);
        ufs.clear(1,n);
        printf("Case %d:\n",times++);
        while(m--)
        {
            scanf("\n%c",&o);
            if(o=='T')
            {
                scanf("%d %d",&u,&v);
                ufs.Union(u,v);
            }
            else
            {
                scanf("%d",&u);
                printf("%d %d %d\n",ufs.query_at(u),ufs.query_cnt(u),ufs.query_move(u));
            }
        }
    }

    return 0;
}