uva 10158 - War(并查集）

题目链接：10158 - War

题目大意：有n个人，若干个命令，每个人分属于不同的国家，并且代表国家与其他代表会谈，这将决定国与国之间的同盟关系，1：a与b结盟  2：a与b决裂（成为敌人） 3：判断a与b是否为同盟（不包括不确定） 4：判断a与b是否为敌人。

注意：同盟的同盟是同盟，敌人的敌人是朋友。添加新的关系时与已有的关系冲突的话输出-1。

解题思路：开一个2 * n的数组，0~n - 1表示的是同盟，n~2 * n - 1表示的是n个国家的敌人。

#include <stdio.h>
#include <string.h>
const int N = 100005;
int n, f[N * 2];

int getfather(int x) {
	return x == f[x] ? x : f[x] = getfather(f[x]);
}

void setfirend(int x, int y) {
	int pf = getfather(x), qf = getfather(y);
	int pe = getfather(x + n), qe = getfather(y + n);

	if (pe == qf || qe == pf) {
		printf("%d\n", -1);
		return;
	}

	f[pf] = qf;
	f[pe] = qe;
}

void setenemies(int x, int y) {
	int pf = getfather(x), qf = getfather(y);
	int pe = getfather(x + n), qe = getfather(y + n);

	if (pf == qf || pe == qe) {
		printf("%d\n", -1);
		return;
	}

	f[pf] = qe;
	f[qf] = pe;
}

void isfirend(int x, int y) {
	int pf = getfather(x), qf = getfather(y);
	printf("%d\n", pf == qf ? 1 : 0);
}

void isenemies(int x, int y) {
	int pf = getfather(x), qe = getfather(y + n);
	printf("%d\n", pf == qe ? 1 : 0);
}

int main () {
	scanf("%d", &n);
	int cas, a, b;
	for (int i = 0; i < n; i++) f[i] = i, f[i + n] = i + n;
	while (scanf("%d%d%d", &cas, &a, &b), cas || a || b) {
		switch(cas) {
			case 1:
				setfirend(a, b);
				break;	
			case 2:
				setenemies(a, b);
				break;
			case 3:
				isfirend(a, b);
				break;
			case 4:
				isenemies(a, b);
				break;
		}
	}
	return 0;
}