POJ1182(权值并查集）

最新推荐文章于 2021-04-21 18:09:22 发布

原创最新推荐文章于 2021-04-21 18:09:22 发布 · 447 阅读

0 ·

CC 4.0 BY-SA版权

数据结构专栏收录该内容

4 篇文章

订阅专栏

本文分享了一种使用并查集解决特定类型问题的编程经验，重点介绍了如何通过并查集进行元素查找和集合合并，以及如何处理相关操作中的边界情况。

解题思路：做到现在的题，还头一次碰上这种并查集，值得纪念。AC的时候用while(scanf("%d%d",&n,&k)!=EOF)读入一直报错，据说是因为只有一组测试数据。推导见下：

#include<cstdio>
#include<cstring>
using namespace std;
int fa[50009],re[50009];
int cd,x1,x2,ans,n;
int find(int x)
{
	int k=fa[x];
	if(fa[x]!=x)
	{
		fa[x]=find(fa[x]);
		re[x]=(re[x]+re[k])%3;
	}
	return fa[x];
}
void unite(int c,int x,int y)
{
	int fax,fay;
	fax=find(x);
	fay=find(y);
	fa[fax]=fay;
	re[fax]=(c-1+3-re[x]+re[y])%3;
}
void solve()
{
	int fax,fay,tp;
	fax=find(x1);
	fay=find(x2);
	if(fax==fay)
	{
		tp=(3-re[x2]+re[x1])%3;
		if(tp!=cd-1)
		{
			ans++;
			return;
		}
	}else
	{
		unite(cd,x1,x2);
	} 
}
int main()
{
	int k;
	//freopen("t.txt","r",stdin);
	//while(scanf("%d%d",&n,&k)!=EOF)
	scanf("%d%d",&n,&k);
	{
		ans=0;
		for(int i=0;i<=n;i++)
		{
			fa[i]=i;
			re[i]=0;
		}
		while(k--)
		{
			scanf("%d%d%d",&cd,&x1,&x2);
			if(cd==2&&x1==x2) 
			{
				ans++;
				continue;
			}
			if(x1>n||x2>n)
			{
				ans++;
				continue;
			}
			solve();
		}
		printf("%d\n",ans);
	}
}