HDU3635Dragon Balls（并查集）

http://acm.hdu.edu.cn/showproblem.php?pid=3635

题目意思是说n个球在n个城市。

每次操作把编号i的球所在的城市的所有的求全部一道另一城市B

每次询问访问编号i的球的城市，以及这个城市的球的数量，以及这个球被移动了多少次。

方法就是用一个结构体记录每个球的父节点和移动次数以及这个节点的球的个数（球和城市可以堪为一个整体），然后每次操作就行。

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <ctime>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
#define INF 0x3f3f3f3f
#define MAX(a,b) (a > b ? a : b)
#define MIN(a,b) (a < b ? a : b)
#define mem0(a) memset(a,0,sizeof(a))
#define lson k<<1, L, mid
#define rson k<<1|1, mid+1, R

typedef long long LL;
const double eps = 1e-12;
const int MAXN = 100005;
const int MAXM = 500005;

struct NODE
{
    int num;
    int fa;
    int time;
}p[MAXN];
int N, Q;

void init()//初始化
{
    for(int i=0;i<=N;i++) {
        p[i].fa = i;
        p[i].num = 1;
        p[i].time = 0;
    }
}

int find(int x)//找父节点
{
    if(x == p[x].fa) return x;
    int fa = p[x].fa;//记录这个节点的父节点
    p[x].fa= find(p[x].fa);
    p[x].time += p[fa].time;//这个节点移动的次数加上他父节点次数
    return p[x].fa;
}

void Union(int x, int y)//合并
{
    int a = find(x);
    int b = find(y);
    if(a != b)
    {
        p[a].fa = p[b].fa;//连边
        p[b].num += p[a].num;//龙珠的数量
        p[a]. time ++;//a又多移动了一次
    }
}

int main()
{
    //freopen("in.txt", "r", stdin);
    int T=0, Cas;
    scanf("%d", &Cas);
    while(Cas--)
    {
        scanf("%d%d%*c", &N, &Q);
        init();
        char ch; int a, b;
        printf("Case %d:
", ++T);
        for(int i=0;i<Q;i++)
        {
            scanf("%c", &ch);
            if(ch == 'T') {
                scanf("%d %d%*c", &a, &b);
                Union(a, b);
            }
            else {
                scanf("%d%*c", &a);
                b = find(a);//必须先找一遍他被移动了多少次
                printf("%d %d %d
", b, p[b].num, p[a].time);
            }
        }
    }
    return 0;
}

但是我把树的层数减少那步去掉后字节搜索层数居然和这个没什么区别，只能说数据太弱了= =

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <ctime>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
#define INF 0x3f3f3f3f
#define MAX(a,b) (a > b ? a : b)
#define MIN(a,b) (a < b ? a : b)
#define mem0(a) memset(a,0,sizeof(a))
#define lson k<<1, L, mid
#define rson k<<1|1, mid+1, R

typedef long long LL;
const double eps = 1e-12;
const int MAXN = 100005;
const int MAXM = 500005;

struct NODE
{
    int num;
    int fa;
}p[MAXN];
int N, Q;

void init()
{
    for(int i=0;i<=N;i++) {
        p[i].fa = i;
        p[i].num = 1;
    }
}

int find(int x)
{
    if(x == p[x].fa) return x;
    return find(p[x].fa);
}

void Union(int x, int y)
{
    int a = find(x);
    int b = find(y);
    if(a != b)
    {
        p[a].fa = b;
        p[b].num += p[a].num;
    }
}

int main()
{
    int T=0, Cas;
    scanf("%d", &Cas);
    while(Cas--)
    {
        scanf("%d%d%*c", &N, &Q);
        init();
        char ch; int a, b;
        printf("Case %d:
", ++T);
        for(int i=0;i<Q;i++)
        {
            scanf("%c", &ch);
            if(ch == 'T') {
                scanf("%d %d%*c", &a, &b);
                Union(a, b);
            }
            else {
                int time = 0;
                scanf("%d%*c", &a);
                while(a != p[a].fa){ time ++; a = p[a].fa; }
                printf("%d %d %d
", a, p[a].num, time);
            }
        }
    }
    return 0;
}