时间限制:1.000秒
你的任务是实现一个并查集的变种,支持如下操作,每组数据共有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;
}
}
}
}