poj 1733 Parity game - 并查集

题目传送门

　　传送点I

　　传送点II

题目大意

　　有一个很长很长的01串（你不知道是什么）。有$n$条线索，每条线索是01串上的一段的1的个数的奇偶性。求最小的$X$，使得存在一个串满足前$X$条线索，但不存在串满足前$X + 1$条线索或者输出$n$表示存在一个串满足所有线索。

　　将区间求和变成前缀和相减，这样每条线索等价于告诉你两个位置的前缀和奇偶性是否相同。

　　直接上带权并查集维护。对于每条线索如果涉及的两个位置没有连通就连通，否则判断是否合法。

Code

/**
 * poj
 * Problem#1733
 * Accepted
 * Time: 32ms
 * Memory: 816k
 */ 
#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <cstdio>
using namespace std;
typedef bool boolean;

const int N = 5e3 + 5;

int L, n;
char str[10];
pair<int, int> ps[N];
boolean relation[N];
int buf[N << 1];
int uf[N << 1], mask[N << 1];

inline void init() {
    scanf("%d%d", &L, &n);
    for (int i = 1; i <= n; i++) {
        scanf("%d%d%s", &ps[i].first, &ps[i].second, str);
        ps[i].first--;
        relation[i] = (str[0] == 'o');
        buf[(i << 1) - 1] = ps[i].first, buf[(i << 1)] = ps[i].second;
    }        
}

void discrete() {
    sort(buf + 1, buf + (n << 1) + 1);
    for (int i = 1; i <= n; i++) {
        ps[i].first = lower_bound(buf + 1, buf + (n << 1) + 1, ps[i].first) - buf;
        ps[i].second = lower_bound(buf + 1, buf + (n << 1) + 1, ps[i].second) - buf;
    }
}

int find(int x) {
    if (uf[x] == x)
        return x;
    int lastfather = uf[x];
    uf[x] = find(uf[x]);
    mask[x] ^= mask[lastfather];
    return uf[x];
}

inline void solve() {
    int doublen = n << 1;
    for (int i = 1; i <= doublen; i++)
        uf[i] = i, mask[i] = 0;
    for (int i = 1; i <= n; i++) {
        if (find(ps[i].first) == find(ps[i].second)) {
            int val1 = mask[ps[i].first], val2 = mask[ps[i].second];
            if ((val1 ^ val2) != relation[i]) {
                printf("%d\n", i - 1);
                return;
            }
        } else {
            int ffather = find(ps[i].first);
            mask[ffather] = relation[i] ^ mask[ps[i].first];
            uf[ffather] = ps[i].second;
        }
    }
    printf("%d\n", n);
}

int main() {
    init();
    discrete();
    solve();
    return 0;
}