UVA11987 Almost Union-Find

思路:

题目中与一般并查集不同的是,要把一个数从一个集合中拿出来,放到另一个集合中

如果现有两个集合 { 1 , 2 } \{1,2\} {1,2} , { 3 } ,\{3\} ,{3}

且p[1]=2

如果要把2拿出来给3

简单的p[2]=3是不行的,因为1也会受到牵连

想法:

可以给数组开$2N$
$1$~$N$储存数字,$N+1$~$2N$储存 减去N后的数字对应的集合
这样可以很方便地拿出单个数字放入别的集合中.原集合中的数进行find操作也不会受影响

#include<stdio.h>
#include<iostream>
#include<cstdlib>
#include<string.h>
#include<algorithm>
using namespace std;
const int N=1e5+10;
int p[N*2];
int size[N*2];
int sum[N*2];
int find(int x)
{
	if(p[x]!=x) p[x]=find(p[x]);
	return p[x];
}
int n,m;
int main()
{
	//freopen("uva.txt","r",stdin);
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		for(int i=1;i<=n*2;i++)
		{
			if(i<=n)
			{
				p[i]=n+i;
			}
			else
			{
				p[i]=i;
				sum[i]=i-n;
				size[i]=1;
			}
		}
		while(m--)
		{
			int op,a,b;
			scanf("%d",&op);
			if(op!=3)
			scanf("%d%d",&a,&b);
			else scanf("%d",&a);
			int pa=find(a);
			int pb=find(b);
			if(op==1)
			{
				if(pa!=pb)
				{
					p[pa]=pb;
					sum[pb]+=sum[pa];
					size[pb]+=size[pa];
				}
			}
			else if(op==2)
			{
				if(pa!=pb)
				{
					sum[pb]+=a;
					size[pb]++;
					
					p[a]=pb;
					
					size[pa]--;
					sum[pa]-=a;
				}
			}
			else
			{
				printf("%d %d\n",size[pa],sum[pa]);
			}
		}
	}
	return 0;
}