题意很简单:1代表合并p、q,2代表将p合并到q,3代表输出p所含节点个数及其节点和
对于删除,设置一个“假父亲”节点(fake数组),使其指向真的原数组,这样删除时只要让fake数组的值改变,这样不会影响别的关系,这是删除操作的核心思想。
#include <iostream>
#include <cmath>
#include <cstdio>
using namespace std;
int maps[100100],fake[100010],sum[100010],cnt[100010];
int getboss(int p){//原并查集数组 假父亲数组 节点和数组 节点个数记录数组
if(maps[p]==p){
return p;
}
else{
return maps[p]=getboss(maps[p]);
}
}
int main(){
ios::sync_with_stdio(0);
int n,m;
while(cin>>n>>m){
for(int i=1;i<=n;i++){
maps[i]=i;
fake[i]=i;//让假父亲指向原数组对应位置
sum[i]=i;
cnt[i]=1;
}
for(int i=0;i<m;i++){
int op,p,q;
cin>>op;
if(op==1){
cin>>p>>q;
int rp=getboss(fake[p]),rq=getboss(fake[q]);//这里得通过fake数组找真正的父亲
if(rp!=rq){
maps[rq]=rp;
sum[rp]+=sum[rq];
cnt[rp]+=cnt[rq];
}
}
if(op==2){
cin>>p>>q;
int rp=getboss(fake[p]),rq=getboss(fake[q]);
if(rp!=rq){
fake[p]=rq;//删除的精要 让fake指向真父亲
cnt[rp]--;
cnt[rq]++;
sum[rp]-=p;
sum[rq]+=p;
}
}
if(op==3){
cin>>p;
int r=getboss(fake[p]);
printf("%d %d\n",cnt[r],sum[r]);
}
}
}
return 0;
}