POJ 2492 - A Bug's Life

Advanced Data Structures :: Disjoint Set

Description

一道灰常和谐的gay问题。

两只小虫走得比较近，就说明他们有关系。

但是如果出现形如三角恋的关系，就说明其中必然有一对gay。

你要做的就是检测小虫子间是否有gay的出现。

Type

Advanced Data Structures :: Disjoint Set

Analysis

利用mod2的加权并查集来解决。

假设每对在一起的小虫子都是异性，

合并前检查两只小虫的性别，查看是否有问题即可。

Solution

// POJ 2492// A Bug's Life// by A Code Rabbit#include <cstdio>#include <cstring>const int MAXN = 2002;const int MOD = 2;struct DisjointSet {    int p[MAXN];    int w[MAXN];    void Init(int);    void Make(int x) { p[x] = x; }    int Find(int x) {        if (p[x] == x) return x;        int res = Find(p[x]);        w[x] = (w[x] + w[p[x]]) % MOD;        return p[x] = res;    }    void Union(int x, int y, int d) {        int px = Find(x); int py = Find(y);        p[px] = py;        w[px] = (w[y] - w[x] + d + MOD) % MOD;    }};void DisjointSet::Init(int n) {    memset(w, 0, sizeof(w));    for (int i = 0; i < n; i++) Make(i);;}int n, m;DisjointSet set;bool is_gay;int main() {    int tot_case;    bool is_first = true;    scanf("%d", &tot_case);    for (int t = 0; t < tot_case; t++) {        scanf("%d%d", &n, &m);        set.Init(n);        is_gay = false;        for (int i = 0; i < m; i++) {            int x, y;            scanf("%d%d", &x, &y);            if (is_gay) continue;            int px = set.Find(x - 1);            int py = set.Find(y - 1);            if (px == py) {                if (set.w[x - 1] == set.w[y - 1])                    is_gay = true;            } else {                set.Union(x - 1, y - 1, 1);            }        }        printf("Scenario #%d:\n", t + 1);        puts(is_gay ? "Suspicious bugs found!" : "No suspicious bugs found!");        puts("");    }    return 0;}