POJ 1733 带权并查集

题意

传送门 POJ 1733 Parity game

题解

观察到可以根据之前的询问结果递推之后的询问结果，考虑使用带权并查集维护。考虑前缀和，那么区间信息可以通过区间端点作差求出，若区间 $[l,r]$ 和为奇数，则 $l-1$ 与 $r$ 连一条权值为 $1$ 的边；反之，连一条权值为 $0$ 的边。此时可以通过异或操作求解节点到根节点的区间内 $1$ 的个数的奇偶性。由于区间过大，将询问的区间端点离线下来离散化处理。

#include <algorithm>
#include <cstdio>
using namespace std;
const int maxn = 5005;
int M, Q, N, L[maxn], R[maxn], F[maxn], sx[maxn * 2], fa[maxn * 2], d[maxn * 2];

inline int cp(int x)
{
    return lower_bound(sx, sx + N, x) - sx;
}

int find(int x)
{
    if (fa[x] == x)
        return x;
    int r = find(fa[x]);
    d[x] ^= d[fa[x]];
    return fa[x] = r;
}

int main()
{
    scanf("%d%d", &M, &Q);
    for (int i = 0; i < Q; ++i)
    {
        char s[5];
        scanf("%d%d%s", L + i, R + i, s);
        F[i] = s[0] == 'e' ? 0 : 1, sx[N++] = L[i] - 1, sx[N++] = R[i];
    }
    sort(sx, sx + N);
    N = unique(sx, sx + N) - sx;
    for (int i = 0; i < N; ++i)
        fa[i] = i;
    int res = Q;
    for (int i = 0; i < Q; ++i)
    {
        int x = cp(L[i] - 1), y = cp(R[i]);
        int q = find(x), p = find(y);
        if (q == p)
        {
            if (F[i] != (d[x] ^ d[y]))
            {
                res = i;
                break;
            }
        }
        else
            fa[q] = p, d[q] = d[x] ^ d[y] ^ F[i];
    }
    printf("%d\n", res);
    return 0;
}