题意:要求你实现一个并查集,有以下3个功能:1,合并两个集合;2,将某个元素移动到另一个集合中;3,求某元素所在集合的元素个数与它们的和。
思路:对于普通并查集,只具备第1个功能。但是仔细研究一下功能3,发现它只查询整个集合的元素个数与和,这个是比较好实现的,只需要维护根结点的元素个数与和就行,无须维护每个父亲结点,因为功能2只能移动一个结点。问题还是在功能2上,要移动一个结点p,首先要删除它,这样会破坏并查集原来的结构,需要寻找它所有的孩子结点以改变它们的父亲,这样做比较麻烦。实际上该结点存在的意义就是连接它的孩子结点与它的父亲结点。这样可以开辟一个新节点id[p]=++now来实现结点的移动,而原来的结点还存在于原来的并查集中,并查集的结构并未改变。数组id[p]用来存放结点p当前实际的id。换句话说,就是开一个数组id【i】表示题中给的元素i当前对应的结点。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <vector>
#include <queue>
#include <map>
#include <algorithm>
#define ll long long
#define INF 2139062143
#define MOD 20071027
#define MAXN 100005
using namespace std;
struct Union_Find
{
int father[MAXN],sum[MAXN],num[MAXN],id[MAXN],cnt;
void init(int n)
{
for(int i=1; i<=n; ++i)
{
father[i]=i;
sum[i]=i;
num[i]=1;
id[i]=i;
}
cnt=n;
}
int findset(int p)
{
return p==father[p]?p:(father[p]=findset(father[p]));
}
void move(int p,int q)
{
int fp=findset(id[p]),fq=findset(id[q]);
if(fp!=fq)
{
num[fp]--;
sum[fp]-=p;
id[p]=++cnt;
num[id[p]]=1;
sum[id[p]]=p;
sum[fq]+=p;
num[fq]++;
father[id[p]]=fq;
}
}
void unio(int p,int q)
{
int fp=findset(id[p]),fq=findset(id[q]);
if(fp!=fq)
{
sum[fq]+=sum[fp];
num[fq]+=num[fp];
father[fp]=fq;
}
}
};
Union_Find un;
int main()
{
int n,m;
while(scanf("%d%d",&n,&m)!=EOF)
{
int p,x,y;
un.init(n);
while(m--)
{
scanf("%d",&p);
if(p==1)
{
scanf("%d%d",&x,&y);
un.unio(x,y);
}
else if(p==2)
{
scanf("%d%d",&x,&y);
un.move(x,y);
}
else
{
scanf("%d",&x);
x=un.id[x];
x=un.findset(x);
printf("%d %d\n",un.num[x],un.sum[x]);
}
}
}
return 0;
}