题意:
初始给定n个集合:{1},{2},…,{n},要求支持三种操作:
"1 p q":若p,q不在同一集合,将它们所在的集合合并成一个
"2 p q":若p,q不在同一集合,将元素p移动到q所在的集合
"3 p" :询问p所在集合的元素个数及元素和
总共m个操作,1<=n,m<=10^5
分析:第一种操作直接合并就可以,第三种操作则开俩个数组,一个记录集合的个数,另一个记录集合的和;
而操作二因为要删除元素,但删除元素对原集合的结构造成影响,所以我们开一个数组来标记他的编号。
删除p元素,改变p的编号,所以原编号p就没有用了,拿新的编号p直接合并就行
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
typedef long long ll;
const int INF=0x3f3f3f3f;
const int N=200000;
int f[N+10],sum[N+10],Size[N+10],p[N+10];
void init()
{
for(int i=0; i<=N; i++)
{
f[i]=p[i]=sum[i]=i;
Size[i]=1;
}
}
int getf(int v)
{
return f[v]==v?v:f[v]=getf(f[v]);
}
int main()
{
int n,m;
while(~scanf("%d%d",&n,&m))
{
init();
int t=n;
while(m--)
{
int a,b,c;
scanf("%d",&a);
if(a==1)
{
scanf("%d%d",&b,&c);
int t1=getf(p[b]);
int t2=getf(p[c]);
if(t1!=t2)
{
sum[t1]+=sum[t2];
Size[t1]+=Size[t2];
f[t2]=t1;
}
}
else if(a==2)
{
scanf("%d%d",&b,&c);
int t1=getf(p[b]);
int t2=getf(p[c]);
if(t1!=t2)
{
sum[t1]-=b;
Size[t1]--;
t1=p[b]=++t;
f[t1]=t2;
sum[t2]+=b;
Size[t2]++;
}
}
else if(a==3)
{
scanf("%d",&b);
int T=getf(p[b]);
printf("%d %d\n",Size[T],sum[T]);
}
}
}
}