【UVA11987】Almost Union-Find

题面

题意

  有$n$个人，有$m$个，三种操作：
  ①.合并$p$所在的集合和$q$所在的集合
  ②.将$p$加入$q$所在的集合
  ③.询问$p$所在的集合中的元素个数以及元素之和

解法

并查集：
  这道题的关键就是②操作
  ①操作很简单，直接使用并查集合并即可
  ③操作可以开一个$siz$数组和一个$sum$数组，在合并$fa$数组时一起合并
  对于②操作，如果直接将$p$删除，然后加入$q$所在的集合，那么$p$所在集合的树结构将会被破坏，所以我们可以给每个点一个编号，一开始编号为自己，如果需要删除$p$，那么就将$rt_{id_p}$的$siz$-1，$sum-p$，然后赋予$p$一个新编号$id_p'$，加入$rt_{id_q}$即可

复杂度

O（$T*nlogn$）

代码

#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<queue>
#define Lint long long int
using namespace std;
const int MAXN=200010;
int f[MAXN],sum[MAXN];
int p[MAXN],siz[MAXN];
int n,m,cnt;
int find(int x)   { return x==f[x] ? f[x] : f[x]=find( f[x] ) ; }
int main()
{
    int opt,u,v,x,y;
    while( scanf("%d%d",&n,&m)!=EOF )
    {
        cnt=n;
        for(int i=1;i<=n;i++)   f[i]=p[i]=sum[i]=i,siz[i]=1;
        for(int i=1;i<=m;i++)
        {
            scanf("%d",&opt);
            if( opt==1 )
            {
                scanf("%d%d",&u,&v);
                u=p[u],v=p[v];
                u=find( u ),v=find( v );
                if( u==v )   continue ;
                f[u]=v;
                siz[v]+=siz[u],sum[v]+=sum[u];
            }
            if( opt==2 )
            {
                scanf("%d%d",&u,&v);
                x=find( p[u] ),y=find( p[v] );
                if( x==y )   continue ;
                sum[x]-=u,siz[x]--;
                x=p[u]=++cnt;
                f[x]=y;
                sum[y]+=u,siz[y]++;
            }
            if( opt==3 )
            {
                scanf("%d",&u);
                u=find( p[u] );
                printf("%d %d\n",siz[u],sum[u]);
            }
        }
    }
    return 0;
}