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[N],id[N];
LL sum[N],cnt[N];
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]=n;
        sum[n]=p;
        cnt[n]=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){
                 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;
}