UVA10158 - War(并查集)

题目链接

题目大意：有n个国家，每个国家有友国和敌国，并且遵守着朋友的朋友是朋友，敌人的敌人是朋友的原则，给你四种操作，题目中有描述，然后让你根据操作进行，对应输出相应的结果。

解题思路：题目里面有敌对关系和朋友关系，朋友的关系可以使用并查集，但是敌对的关系是不行的。转化敌对的关系。假设0～n - 1，代表国家。n～2 * n - 1 代表0～n - 1这些国家的敌国。例如 1 和 4是敌对关系，因为敌人的敌人是朋友，那么1 + n 和 4 与 1 和4 + n就是朋友的关系，将敌对的关系转换成朋友的关系，然后朋友的关系就可以用并差集了（朋友具有传递的关系）。

代码：

#include <cstdio>
#include <cstring>const int maxn = 10005;
int N;
int p[maxn * 2];void init () {for (int i = 0; i < 2 * N; i++)p[i] = i;
}int getParent(int x) {return x == p[x] ? x : p[x] = getParent(p[x]);
}bool setfriend (int x, int y) {int  x1 = getParent(x);int  y1 = getParent(y);int  x2 = getParent(x + N);int  y2 = getParent(y + N);if (x1 == y2 || y1 == x2)return false;else {p[x1] = y1;p[x2] = y2;return true;}
}bool setenemies(int x, int y ) {int  x1 = getParent(x);int  y1 = getParent(y);int  x2 = getParent(x + N);int  y2 = getParent(y + N);if (x1 == y1 || y2 == x2)return false;else {p[x1] = y2;p[y1] = x2;return true;}
}bool is_friend(int x, int y) {int  x1 = getParent(x);int  y1 = getParent(y);int  x2 = getParent(x + N);int  y2 = getParent(y + N);if (x1 == y1)return true;return false;
}bool is_enemies(int x, int y) {int  x1 = getParent(x);int  y1 = getParent(y);int  x2 = getParent(x + N);int  y2 = getParent(y + N);if (x1 == y2 || x2 == y1)return true;return false;
}int main () {int op, x, y;scanf ("%d", &N);init();while (scanf ("%d%d%d", &op, &x, &y) != EOF && (op || x || y)) {if (op == 1) {if (setfriend(x, y) == false)printf("-1\n");} else if (op == 2) {if (setenemies(x, y) == false)printf ("-1\n");} else if (op == 3)printf ("%d\n", is_friend(x, y));else printf ("%d\n", is_enemies(x, y));}return 0;
}