POJ1182食物链 并查集

Description

动物王国中有三类动物A,B,C，这三类动物的食物链构成了有趣的环形。A吃B， B吃C，C吃A。 
现有N个动物，以1－N编号。每个动物都是A,B,C中的一种，但是我们并不知道它到底是哪一种。 
有人用两种说法对这N个动物所构成的食物链关系进行描述： 
第一种说法是"1 X Y"，表示X和Y是同类。 
第二种说法是"2 X Y"，表示X吃Y。 
此人对N个动物，用上述两种说法，一句接一句地说出K句话，这K句话有的是真的，有的是假的。当一句话满足下列三条之一时，这句话就是假话，否则就是真话。 
1） 当前的话与前面的某些真的话冲突，就是假话； 
2） 当前的话中X或Y比N大，就是假话； 
3） 当前的话表示X吃X，就是假话。 
你的任务是根据给定的N（1 <= N <= 50,000）和K句话（0 <= K <= 100,000），输出假话的总数。 

Input

第一行是两个整数N和K，以一个空格分隔。 
以下K行每行是三个正整数 D，X，Y，两数之间用一个空格隔开，其中D表示说法的种类。 
若D=1，则表示X和Y是同类。 
若D=2，则表示X吃Y。

Output

只有一个整数，表示假话的数目。

Sample Input

100 7
1 101 1 
2 1 2
2 2 3 
2 3 3 
1 1 3 
2 3 1 
1 5 5

Sample Output

3

 

分析：

方法1：种类并查集。将确定关系的元素合并到同一集合中，利用rela数组记录各节点与父节点的关系，其中0表示同一类，1表示吃父节点，2表示被父节点吃。

更新结点关系时，可利用向量的方法计算，rela[x]即为向量x->fx，x与新父结点关系rela[x]=(rela[x]+rela[fx])%3（向量加法）。

由给定条件d可以得出向量x->y的值d-1。若x和y已属同一集合z，则关系已经确定，由向量公式x->y=x->z - y->z判断是否为假话；否则合并集合,将x所属集合并入y所属集合，x父节点fx与y父节点fy的关系fx->fy由x->fx,x->y,y->fy得到：

rela[fx]=(3-rela[x]+kind+rela[y])%3。

#include<iostream>
#include<cstring>
using namespace std;
const int maxn=50005; 
int fa[maxn],rela[maxn];//0同类，1吃父节点，2被父节点吃 

void init(int n)
{
	for(int i=1;i<=n;i++)
	{
		fa[i]=i;
		rela[i]=0;
	}
}

int find(int x)
{
	if(fa[x]!=x)
	{
		int y=fa[x];
		fa[x]=find(y);
		rela[x]=(rela[x]+rela[y])%3;//更新x与根节点关系 
	}
	return fa[x];
} 

void unite(int x,int y,int kind)
{
	int fx=find(x),fy=find(y);
	if(fx==fy) return;
	fa[fx]=fy;
	rela[fx]=(3-rela[x]+kind+rela[y])%3;//更新fx与fy关系 
} 

bool same(int x,int y)
{
	return find(x)==find(y);
}

int main()
{
	int n,k,ans=0;
	cin>>n>>k; 
	init(n); 
	while(k--)
	{
		int d,x,y,kind;
		scanf("%d%d%d",&d,&x,&y);
		if(x>n||y>n)
		{
			ans++;
			continue;				
		}
		kind=d-1;
		if(same(x,y))
		{//如果已确定关系 
			if(kind!=(rela[x]-rela[y]+3)%3)//通过根节点判断x与y关系 
				ans++;
		}
		else
		{
			unite(x,y,kind);
		}
	}
	cout<<ans<<endl;
	return 0;
} 

 

方法2：定义并查集:i-x表示“i属于种类x”，并查集里的每一个组表示组内所有元素代表的情况同时发生或不发生。对每个动物创建3个元素i-A,i-B,i-C,共3*N个元素。

对于x,y属于同一类，合并x-A和y-A，x-B和y-B，x-C和y-C。对于x吃y，合并x-A和y-B，x-B和y-C，x-C和y-A。

合并前先进行判断。第一类信息需判断x-A和y-B或y-C是否在同一组。第二类要判断x-A和y-A或y-C是否在同一组。

#include<iostream>
#include<cstring>
using namespace std;
const int maxn=50005; 
int fa[maxn*3],rank[maxn*3];

void init(int n)
{
	for(int i=1;i<=3*n;i++)
	{
		fa[i]=i;
		rank[i]=1;
	}
}

int find(int x)
{
	if(fa[x]!=x)
	{
		fa[x]=find(fa[x]);
	}
	return fa[x];
} 

void unite(int x,int y)
{
	int fx=find(x),fy=find(y);
	if(fx==fy) return;
	if(rank[fx]>rank[fy])
	{
		fa[fy]=fx;
	}
	else
	{
		fa[fx]=fy;
		if(rank[fx]=rank[fy])
			rank[fy]++;
	}
} 

bool same(int x,int y)
{
	return find(x)==find(y);
}

int main()
{
	int n,k,ans=0;
	cin>>n>>k; 
	init(n); 
	while(k--)
	{
		int d,x,y;
		scanf("%d%d%d",&d,&x,&y);
		if(x>n||y>n)
		{
			ans++;
			continue;				
		}
		if(d==1)
		{
			if(same(x,y+n)||same(x,y+2*n))
			{//如果已确定x吃y或y吃x，则为假话 
				ans++;
				continue;
			}
			unite(x,y);unite(x+n,y+n);unite(x+2*n,y+2*n);
		}
		else
		{
			if(same(x,y)||same(x,y+2*n))
			{//如果已确定x和y同类或y吃x，则为假话 
				ans++;
				continue;
			}
			unite(x,y+n);unite(x+n,y+2*n);unite(x+2*n,y);
		}
	}
	cout<<ans<<endl;
	return 0;
}