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），输出假话的总数。 

输入

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

输出

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

样例输入

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

样例输出

3

经典的并查集题目。

对于每只动物我们构造三个种类，A,B,C，我们可以用并查集维护种类。

x 代表种类A ；x+N代表种类B ；x+2*N 代表种类C

y 代表种类B ; y+N代表种类B ; y+2*N 代表种类C 

所以就按照如下操作就可以了

当D = 1 ，代表x,y属于同于种类----合并(x,y), (x+N,y+N), (x+2*N,y+2*N)

当D = 2 ，代表x吃y--------------合并(x,y+N), (x+N,y+2*N), (y,x+2*N)

具体代码如下：

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
int N, K;
int f[150005];
struct node {
	int D, X, Y;
}s[100005];
void init(int n) {
	for(int i = 0; i < n; i++) {
		f[i] = i;
	}
}

int getf(int v) {
	if(f[v] != v) {
		f[v] = getf(f[v]);
	}
	return f[v];
}

void merge(int v,int u) {
	int t1 = getf(v);
	int t2 = getf(u);
	if(t1 != t2) {
		f[t1] = t2;
	}
}

bool same(int v, int u) {
	return getf(v) == getf(u);
}
bool islimit(int x, int y) {
	return x>=0 && y>=0 && x<N && y<N;
}
/*
1 A B 
A , B
A+N , B+N
A+2*N , B+2*N
2 A B
A , B+N
A+N , B+2*N
A+2*N , B
*/
int main() {
	int ans = 0;
	scanf("%d%d",&N,&K);
	init(3*N);
	for(int i = 0; i < K; i++) {
		scanf("%d%d%d",&s[i].D,&s[i].X,&s[i].Y);
		s[i].X--;s[i].Y--;
	}
	for(int i = 0; i < K; i++) {
		int op = s[i].D;
		int x = s[i].X;
		int y = s[i].Y;
		if(!islimit(x , y)) {
			ans++;
			continue;
		}
		if(op == 1) {
			if(same(x, y+N) || same(x, y+2*N)) {
				ans++;
			}
			else {
				merge(x, y);
				merge(x + N, y + N);
				merge(x + 2*N, y + 2*N);
			}
		}
		else if(op == 2) {
			if(same(x, y) || same(x, y + 2*N)) {
				ans++;
			}
			else {
				merge(x, y + N);
				merge(x + N, y + 2*N);
				merge(x + 2*N, y);
			}
		}
	}
	printf("%d\n",ans);
	return 0;
}