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;
}