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