wikioi 1074 食物链 并查集

题目描述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 Description

第一行是两个整数N和K，以一个空格分隔。 　　

以下K行每行是三个正整数D，X，Y，两数之间用一个空格隔开，其中 D 表示说法的种类。 　　

若D=1，则表示X和Y是同类。 　　

若D=2，则表示X吃Y。

输出描述Output Description

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

样例输入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

数据范围及提示Data Size & Hint

输入文件  

 对7句话的分析 100 7

1 101 1　　假话

2 1 2　　  真话

2 2 3　　  真话

2 3 3　　  假话

1 1 3　　  假话

2 3 1　　  真话

 1 5 5　　  真话

NOI 2001 食物链(eat)

和上一道题类似，主要贴一下自己的一些想法：

A吃B，B吃C，C吃A，如果用x表示A，x+n表示B，x+2n表示C。那么当告诉1 1 3时，动物1和动物3是同类，可能的情况是1,3都是A，1,3都是B或者1,3都是C，所以要merge( 1,3 ),merge(1+n,3+n),merge(1+2n,3+2n)。

当告诉2 2 3时，动物2吃动物3，可能的情况是2是A,B or C，所以：merge(2,3+n),merge(2+n,3+2n),merge(2+2n,3)。当i和j+n在一个集合中时，说明动物i要吃动物j；当i+2n和j在同一个集合时，同样说明i要吃j。以下附上这种思路的代码：

#include<cstdio>

int n,k;
int f[50000*3+10];

int getroot(int x) { return f[x]==x ? x : f[x]=getroot(f[x]); }
inline void merge(int x,int y) { f[getroot(x)]=getroot(y); }

int main()
{
	freopen("input.txt","r",stdin);
	//freopen("1074.out","w",stdout);
	scanf("%d%d",&n,&k);
	for(int i=1;i<=3*n;i++) f[i]=i;
	int res=0;
	while(k--)
	{
		int d,x,y;
		scanf("%d%d%d",&d,&x,&y);
		if(x>n||y>n) res++;
		else if(d==2 && x==y) res++;
		else if(d==1)//the same
		{
			if(getroot(x)==getroot(y+n)||getroot(x+n)==getroot(y+n+n)||getroot(x+n+n)==getroot(y)) res++;//x eat y
			else if(getroot(x+n)==getroot(y)||getroot(x+n+n)==getroot(y+n)||getroot(x)==getroot(y+n+n)) res++;//y eat x
			else{
				merge(x,y);
				merge(x+n,y+n);
				merge(x+n+n,y+n+n);
			}
		}
		else{//x eat y
			if(getroot(x)==getroot(y)||getroot(x+n)==getroot(y+n)||getroot(x+n+n)==getroot(y+n+n)) res++;//x y are the same
			else if(getroot(x+n)==getroot(y)||getroot(x+n+n)==getroot(y+n)||getroot(x)==getroot(y+n+n)) res++;//x eat y
			else{
				merge(x,y+n);
				merge(x+n,y+n+n);
				merge(x+n+n,y);
			}
		}
	}
	printf("%d\n",res);
	return 0;
}

另外，上面代码可以精简，因为：

else{
	merge(x,y);
	merge(x+n,y+n);
	merge(x+n+n,y+n+n);
}

另外一种喜闻乐见的思路是：不使用偏移向量法枚举，而是多用一个数组kind维护0,1,2三个值。当x吃y时，kind[x]比kind[y]大1，同时father[y]充当father[x]的father，即father[father[x]]=father[y]；当x与y同类时，kind[x]比kind[y]大0，同样要维护成同一个集合，即father[father[x]]=father[y]。这种思路需要注意的问题是如何维护压缩路径时father的修改，如以下代码所示：

int find(int x){
	if(x!=father[x]){
		int tmp=father[x];
		father[x]=find(father[x]);
		kind[x]=(kind[x]+kind[tmp])%3;//顺序不能写反
	}
	return father[x];
}

由于kind[x]总要比 kind[father[x]]大kind[x]，因此发生路径压缩时，相当于更换了father[x]，因此要更新kind[x]，即

kind[x]=(kind[x]+kind[tmp])%3

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<vector>
#include<algorithm>
#define MAXN 50001

using namespace std;
int n,m,p,x,y;
int father[MAXN];
int kind[MAXN];
int cnt;


int find(int x){
	if(x!=father[x]){
		int tmp=father[x];
		father[x]=find(father[x]);
		kind[x]=(kind[x]+kind[tmp])%3;//顺序不能写反
	}
	return father[x];
}

int main(){
	
	//freopen("F:\\练习\\test\\test\\input.txt","r",stdin);

	scanf("%d%d",&n,&m);

	for(int i=1;i<=n;i++) father[i]=i;
	while(m--){
		int d,x,y;
		scanf("%d%d%d",&d,&x,&y);
		if(x>n||y>n){cnt++;continue;}
		if(d==1){
			if( find(x)==find(y) )
				if( kind[x]!=kind[y] ) cnt++;
				else;
			else{
				kind[ father[x] ]=( kind[y]-kind[x]+3 )%3;
				father[father[x]]=father[y];
			}
		}
		else{
			if(x==y){cnt++;continue;}
			if( find(x)==find(y) )
				if( kind[x]!=(kind[y]+1)%3 ) cnt++;
				else;
			else{

				kind[father[x]]=(kind[y]-kind[x]+4)%3;
				father[father[x]]=father[y];
			}
		}
	}
	printf("%d\n",cnt);
	
	return 0;
}