并查集可以做,只不过需要些技巧,昨天写了一下,发现有些问题,合并的话还是比较简单的,直接将他们所在集合的各个值合并即可,但是将p移动的时候就有问题了,当parents[p]不等于p的时候没有什么问题,但是p如果等于parents[p]就有问题了,如果将p移走,那么查找parents[q]=p的节点就会出现问题。我想了很久,没想到什么好办法,今天问了问其他少侠,得到一个不错的思路,就是为每个点设置一个虚拟的parents,这个parents不会被移走,这样的话更新起来就没有顾忌了,改完以后直接A掉了。。。果然群众的力量是无比强大的。
代码:
#include <iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-9
#define pi acos(-1.0)
using namespace std;
const int maxn=200000+10;
const int base=100000;
int parents[maxn],sum[maxn],num[maxn];
int Find(int x)
{
if(x!=parents[x])
{
parents[x]=Find(parents[x]);
}
return parents[x];
}
void Uion(int x,int y)
{
int a=Find(x);
int b=Find(y);
if(a!=b)
{
parents[b]=a;
sum[a]+=sum[b];
sum[b]=0;
num[a]+=num[b];
num[b]=0;
}
}
void Move(int x,int y)
{
int a=Find(x);
int b=Find(y);
if(a!=b)
{
parents[x]=b;
num[a]--;
num[b]++;
sum[b]+=x;
sum[a]-=x;
}
}
void Init(int n)
{
for(int i=0;i<=n;++i)
{
parents[i]=i+base;
parents[i+base]=i+base;
sum[i+base]=i;
num[i+base]=1;
}
}
int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
int n,m;
while(cin>>n>>m)
{
Init(n);
int type,p,q;
for(int i=0;i<m;++i)
{
cin>>type;
if(type==1)
{
cin>>p>>q;
Uion(p,q);
}
else if(type==2)
{
cin>>p>>q;
Move(p,q);
}
else
{
cin>>p;
cout<<num[Find(p)]<<" "<<sum[Find(p)]<<endl;
}
}
}
return 0;
}