【并查集】集合 set

集合

时限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;
}