BZOJ 2821作诗(Poetize) 分块

Description

有一个长度为n的序列，序列每个元素的范围[1,c]，有m个询问x y，表示区间[x,y]中出现正偶数次的数的种类数。

Solution

大力分块解决问题。

把序列分块，f[i][j]表示第i块到第j块的答案，并记录块的前缀数的出现次数。

f[i][j]直接暴力算，块的前缀数的出现次数也可以直接算，都是nsqrt(n)。

遇到询问x y，中间答案的块可以直接统计，然后再暴力统计左右两边零碎的贡献，也是nsqrt(n)。

Code

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>

using namespace std;

#define REP(i, a, b) for (int i = (a), i##_end_ = (b); i <= i##_end_; ++i)
const int maxn = 1e5+10;
int n, c, m, a[maxn];
int bel[maxn], l[maxn], r[maxn];
int cnt[350][maxn], s_block, t[maxn], f[350][350];

int solve(int x, int y)
{
    int L = bel[x], R = bel[y];
    if (L == R)
    {
        REP(i, x, y) t[a[i]] = 0;
        int ret = 0;
        REP(i, x, y) if (t[a[i]] ++) ret += (t[a[i]]&1) ? -1 : 1;
        return ret;
    }
    else
    {
        int ret = f[L+1][R-1];
        REP(i, x, r[L]) t[a[i]] = 0;
        REP(i, l[R], y) t[a[i]] = 0;
        REP(i, x, r[L]) if (!t[a[i]]) t[a[i]] = cnt[R-1][a[i]]-cnt[L][a[i]];
        REP(i, l[R], y) if (!t[a[i]]) t[a[i]] = cnt[R-1][a[i]]-cnt[L][a[i]];
        REP(i, x, r[L]) if (t[a[i]] ++) ret += (t[a[i]]&1) ? -1 : 1;
        REP(i, l[R], y) if (t[a[i]] ++) ret += (t[a[i]]&1) ? -1 : 1;
        return ret;
    }
}

int main()
{
    scanf("%d %d %d", &n, &c, &m);
    REP(i, 1, n) scanf("%d", &a[i]);
    int block = int(sqrt(n));
    REP(i, 1, n)
    {
        bel[i] = i/block+1, r[bel[i]] = i;
        if (i == 1 || bel[i] != bel[i-1]) l[bel[i]] = i;
    }
    s_block = n/block+1;
    REP(i, 1, s_block)
    {
        REP(j, 1, c) t[j] = 0;
        REP(j, i, s_block)
        {
            f[i][j] = (j == i) ? 0 : f[i][j-1];
            REP(k, l[j], r[j]) if (t[a[k]] ++) f[i][j] += (t[a[k]]&1) ? -1 : 1;
        }
    }
    REP(i, 1, s_block)
    {
        REP(j, 1, c) cnt[i][j] = cnt[i-1][j];
        REP(j, l[i], r[i]) cnt[i][a[j]] ++;
    }
    int x, y, ans = 0;
    while (m --)
    {
        scanf("%d %d", &x, &y);
        x = (x+ans)%n+1, y = (y+ans)%n+1;
        if (x > y) swap(x, y);
        printf("%d\n", ans = solve(x, y));
    }
    return 0;
}