UVa 11987 - Almost Union-Find

时间限制：1.000秒

题目链接：http://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&category=24&page=show_problem&problem=3138

　　你的任务是实现一个并查集的变种，支持如下操作，每组数据共有m个操作：

1 p q：合并p和q所在集合，如果已经在一个集合中，忽略此命令
2 p q：把p移动到q所在集合，如果已经在一个集合中，忽略此命令
3 p：输出p所在集合的元素个数和该集合所有元素之和

　　初始的时候共有n个单元素集合：{1}，{2}，…，{n}。

　　题目的名字叫几乎就是并查集，其实还是并查集。只不过这次要把信息附加到根结点上，让根结点记录这棵树内所有元素的个数和所有元素的和，移动时维护根节点信息即可。

　　按照上面的思路，操作1和3并不难实现。

　　接下来就是重点问题：操作2怎么实现。乍一想，直接pa[p] = q不就行了？但问题是如果p下面还有子结点，那么整棵子树就被移走了，而操作2要求的是只移动这一个元素。

　　我这里用的办法是一开始为每一个集合分别创建一个”虚拟“结点，让这些新创建的结点作为根节点，将树的信息放在这些根节点上。这样，进行操作1的时候，由于修改的只是这些在1~n之外的“虚拟“结点，因此1~n的结点之下并不会有子结点。同时，进行操作2的时候，让待移动的结点指向新集合的根结点，而根结点一定是”虚拟“结点，保证了结果的正确性。

#include 
   
   
    
    

const int MAXN = 200000 + 10;

int pa[MAXN], num[MAXN], sum[MAXN];
int n, m, cmd, p, q;

inline void init() {
    for(int i = 0, j = n + 1; i <= n; ++i, ++j) pa[i] = j, pa[j] = j, num[j] = 1, sum[j] = i;
}
int findp(const int &x) {
    return pa[x] == x ? x : pa[x] = findp(pa[x]);
}

inline void cmd1(const int &p, const int &q) {
    int x = findp(p), y = findp(q);
    if(x == y) return;
    else {
        pa[x] = y;
        num[y] += num[x], sum[y] += sum[x];
    }
}
inline void cmd2(const int &p, const int &q) {
    int x = findp(p), y = findp(q);
    if(x == y) return;
    else {
        pa[p] = y;
        num[x] -= 1, sum[x] -= p;
        num[y] += 1, sum[y] += p;
    }
}
inline void cmd3(const int &p) {
    int x = findp(p);
    printf("%d %d\n", num[x], sum[x]);
}


int main() {
    while(~scanf("%d%d", &n, &m)) {
        init();
        for(int i = 0; i != m; ++i) {
            scanf("%d", &cmd);
            switch(cmd) {
            case 1:
                scanf("%d%d", &p, &q);
                cmd1(p, q);
                break;
            case 2:
                scanf("%d%d", &p, &q);
                cmd2(p, q);
                break;
            case 3:
                scanf("%d", &p);
                cmd3(p);
                break;
            }
        }
    }
}