UVA11987_Almost Union-Find(并查集删除操作)

传送门

题意：一共有n个元素的组合1,2,3….n

现在有三个操作

1 p q：把p,q所在的集合合并

2 p q：把p移动到q所在的集合

3 p：输出p所在的集合的元素的个数和所有元素和

思路： 在基础的并查集下可以轻松完成1，3操作，对于2操作，一开始想直接把p移到q集合去，发现一个问题，如果p是这个集合的root，那他的儿子也会跟过去，所有想一个办法，开一个数组id，如果p，q不是同一个集合，那么我们就先把p这个集合的sum，cnt维护掉，然后把p放在一个新的位置（++n），更新sum，cnt，Union这两个。这样的话我们可以想一想，对于原来的p的集合，即使它是root，我们孩子搜到他的时候发现其sum，cnt是没有问题的，（相当于这个root是一个空壳）。

#include<bits/stdc++.h>
#define debug(a) cout << #a << " " << a << endl
#define LL long long
#define ull unsigned long long
#define PI acos(-1.0)
#define eps 1e-6
const int N=2e5+107;
using namespace std;
int pre
,id
;
LL sum
,cnt
;
int n,m;
void init()
{
for(int i=1;i<N;i++) pre[i]=i,id[i]=i,cnt[i]=1,sum[i]=i;
}
int findx(int x)
{
return pre[x]==x ? x : pre[x]=findx(pre[x]);
}
void Union(int p,int q)
{
int fx=findx(p);
int fy=findx(q);
if(fx!=fy){
pre[fx]=fy;
sum[fy]+=sum[fx];
cnt[fy]+=cnt[fx];
}
}
void Delete(int p,int q)
{
int fx=findx(id[p]);
int fy=findx(id[q]);
if(fx!=fy){
sum[fx]-=p;
cnt[fx]-=1;
id[p]=++n;
pre
=n;
sum
=p;
cnt
=1;
}
}

int main ()
{
//yyy_3y
//freopen("1.in","r",stdin);
while(~scanf("%d%d",&n,&m)){
init();
int op,p,q;
for(int i=1;i<=m;i++){
scanf("%d",&op);
if(op==1){
scanf("%d%d",&p,&q);
Union(id[p],id[q]);
}else if(op==2){

4000
scanf("%d%d",&p,&q);
Delete(p,q);
Union(id[p],id[q]);
}else {
scanf("%d",&p);
p=findx(id[p]);
printf("%lld %lld\n",cnt[p],sum[p]);
}
}
}
return 0;
}