集合
时限1s
set.pascal/c/cpp
【题目描述】
维护一个数据结构支持下列操作:
1 p q 合并包含元素p的集合和包含元素q的集合,如果p和q在同一集合内无视这个操作;
2 p q 把元素p移动到包含元素q的集合,如果p和q在同一集合内无视这个操作;
3 p 输出包含元素p的集合的元素个数和该集合所有元素之和
初始时每个元素属于独立集合
【输入】
输入的第一行是n和m,分别表示元素个数和操作个数。
接下来的m行,每行有一个操作,格式如上所述。
【输出】
输出第三种操作的输出,每个操作输出独立一行。
【输入样例】
5 7
1 1 2
2 3 4
1 3 5
3 4
2 4 1
3 4
3 3
【输出样例】
3 12
3 7
2 8
【数据范围】
对于30%的数据,1<=n<=1000,1<=m<=1000
对于30%的数据不包含第二种操作
对于100%的数据,1<=n<=100000,1<=m<=100000
说实话,1、3操作很好写,就不说了
至于2操作,涉及到了并查集的删除,如果是叶子节点的话就很好处理,直接将其 f [ x ] 指向新的父亲再更新值即可,然而如果是根的话,我们就需要处理其所有儿子的父亲,这样就会以O(N)的代价完成,最终会是一个接近O(N*N)的时间复杂度,会超时
一个很巧妙的方法可以避免我们要改变的 q 为根节点的问题,我们在初始化的时候全部初始化成一个不存在的点,即可以写成 f [ i ] = i + n ; 这样由于有路径压缩,所以我们要改变一个点 q 时,它肯定是叶子节点的,这样处理起来就很方便了
这一题的难点也就到此结束了~
C++ Code
/*
C++ Code
http://blog.csdn.net/jiangzh7
By Jiangzh
*/
#include<cstdio>
const int MAXN=100000+10;
int n,m;
int f[MAXN*2];
int sum[MAXN*2],num[MAXN*2];
int getroot(int x)
{
if(f[x]==x) return x;
return f[x]=getroot(f[x]);
}
void merge(int x,int y)
{
int rx=getroot(x),ry=getroot(y);
num[ry]+=num[rx];sum[ry]+=sum[rx];
f[rx]=ry;
}
void change(int x,int y)
{
int rx=getroot(x),ry=getroot(y);
num[rx]--;sum[rx]-=x;
num[ry]++;sum[ry]+=x;
f[x]=ry;
}
void work()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
f[i]=f[i+n]=i+n;
sum[i]=sum[i+n]=i;
num[i]=num[i+n]=1;
}
int op,p,q;
for(int i=1;i<=m;i++)
{
scanf("%d",&op);
if(op==1)
{
scanf("%d%d",&p,&q);
if(getroot(p)==getroot(q)) continue;
merge(p,q);
}
else if(op==3)
{
scanf("%d",&p);
printf("%d %d\n",num[getroot(p)],sum[getroot(p)]);
}
else if(op==2)
{
scanf("%d%d",&p,&q);
if(getroot(p)==getroot(q)) continue;
change(p,q);
}
}
}
int main()
{
freopen("set.in","r",stdin);
freopen("set.out","w",stdout);
work();
return 0;
}