1. 程式人生 > >POJ 1182 食物鏈(並查集)

POJ 1182 食物鏈(並查集)

動物王國中有三類動物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
挑戰上的思路: 由於N和K很大,所以必須高效地維護動物之間的關係,並快速判斷是否產生了矛盾。並查集是維護 “屬於同一組” 的資料結構,但是在本題中,並不只有屬於同一類的資訊,還有捕食關係的存在。因此需要開動腦筋維護這些關係。
對於每隻動物i建立3個元素i-A, i-B, i-C, 並用這3*N個元素建立並查集。這個並查集維護如下資訊:
① i-x 表示 “i屬於種類x”。
②並查集裡的每一個組表示組內所有元素代表的情況都同時發生或不發生。
例如,如果i-A和j-B在同一個組裡,就表示如果i屬於種類A那麼j一定屬於種類B,如果j屬於種類B那麼i一定屬於種類A。因此,對於每一條資訊,只需要按照下面進行操作就可以了。
1)第一種,x和y屬於同一種類———合併x-A和y-A、x-B和y-B、x-C和y-C。
2)第二種,x吃y—————————合併x-A和y-B、x-B和y-C、x-C和y-A。
#include<cstdio>
const int N = 50010;
int f[3*N];
int Find(int x)
{
	return f[x]==x?x:f[x]=Find(f[x]);
}
void merge(int x,int y)
{
	int xx=Find(x);
	int yy=Find(y);
	if(xx!=yy)
	f[xx]=yy;
}
bool same(int a,int b)
{
	return Find(a)==Find(b);
}
void init()
{
	for(int i=1;i<=3*N;i++)
	f[i]=i;
}
int main()
{
	int n,m;
	int ans=0;
	int x,y,op;
	init();
	scanf("%d%d",&n,&m);
	while(m--)
	{
		scanf("%d%d%d",&op,&x,&y);
		if(x>n||y>n)
		{
			ans++;
			continue;
		}
		if(op==1)
		{
			if(same(x,y+N)||same(y,x+N))//存在 吃或者被吃關係 
			{
				ans++;
			}
			else
			{
				merge(x,y);
				merge(x+N,y+N);
				merge(x+2*N,y+2*N);//會形成“森林” 
			}
		}
		else
		{
			if(same(x,y)||same(y,x+N))//已經是同類 或是被吃關係 
			ans++;
			else
			{
				merge(x,y+N);
				merge(x+N,y+2*N);
				merge(x+2*N,y);
			}
		}
		
	}
	printf("%d\n",ans);
	
}