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+;
 int n, c, m, a[maxn];
 int bel[maxn], l[maxn], r[maxn];
 int cnt[][maxn], s_block, t[maxn], f[][];

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

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