UVA 11987 Almost Union-Find 并查集的删除操作 懒惰标记

你的任务是实现一个并查集的变种。

1 q p 合并 q，p 所在 集合。

2 q p 把 q 移动到 p 所在集合。

3 p  查询 p所在集合的元素个数与元素和。

一开始想使用set 保留 每个并查集的结构，但容易tle。

因为只统计根节点的answer，把删除节点保留下来不影响answer，那么可以新建立一个节点表示 p，放入集合。

#include<iostream>
#include<cstdio>
using namespace std;
#define Size 100005
int f[Size*2],r[Size],n,m,id[Size],ans;
long long s[Size];
int find(int x)
{
	return x==f[x]?x:find(f[x]);
}
void join(int x,int y)
{
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy)
{
	s[fx] +=s[fy];
	r[fx] +=r[fy];
	f[fy]=fx;
}
}
int main()
{
	int q,p,x;
	while(~scanf("%d%d",&n,&m))
	{
		ans =Size;
		for(int i=1;i<=n;i++) f[i]=i,r[i]=1,s[i]=i,id[i]=i;
		while(m--)
		{
			scanf("%d",&x);
			if(x!=3)
			{
				scanf("%d%d",&p,&q);
				int p1=p,q1=q;
			    p=id[p];q=id[q];
				int fp=find(p);int fq=find(q);
				if(x==1) join(p,q);
				if(x==2) 
				{
				  if(fp!=fq)
				  {
			       ans++;
				   id[p1]=ans;
		           r[fp]--;s[fp] -=p1;
		           r[fq]++;s[fq] +=p1;
		           f[ans] =fq;
		          }
				}
			}
			else 
			{
				scanf("%d",&p);
				p=id[p];
				printf("%d %lld\n",r[find(p)],s[find(p)]); 
			} 
	    }
	} 
	return 0; 
}