BZOJ 2821 作诗 分块

BZOJ 2821 作诗

Description

神犇SJY虐完HEOI之后给傻×LYD出了一题：SHY是T国的公主，平时的一大爱好是作诗。由于时间紧迫，SHY作完诗
之后还要虐OI，于是SHY找来一篇长度为N的文章，阅读M次，每次只阅读其中连续的一段[l,r]，从这一段中选出一
些汉字构成诗。因为SHY喜欢对偶，所以SHY规定最后选出的每个汉字都必须在[l,r]里出现了正偶数次。而且SHY认
为选出的汉字的种类数（两个一样的汉字称为同一种）越多越好（为了拿到更多的素材！）。于是SHY请LYD安排选
法。LYD这种傻×当然不会了，于是向你请教……问题简述：N个数，M组询问，每次问[l,r]中有多少个数出现正偶
数次。

Input

输入第一行三个整数n、c以及m。表示文章字数、汉字的种类数、要选择M次。第二行有n个整数，每个数Ai在[1, c
]间，代表一个编码为Ai的汉字。接下来m行每行两个整数l和r，设上一个询问的答案为ans(第一个询问时ans=0)，
令L=(l+ans)mod n+1, R=(r+ans)mod n+1，若L>R，交换L和R，则本次询问为[L,R]。

Output

输出共m行，每行一个整数，第i个数表示SHY第i次能选出的汉字的最多种类数。

Sample Input

5 3 5
1 2 2 3 1
0 4
1 2
2 2
2 3
3 5

Sample Output

2
0
0
0
1

HINT

对于100%的数据，1<=n,c,m<=10^5

---

做这道题最大的收获就是打破了一个观念：分块可以做的题，用线段树之类的数据结构一定可以做。

如果不强制在线，那么这道题就比较水了，思路可以参照这两道题的想法：【河北OI 2012 DAY1】采花，区间MEX。这道题强制在线，但是还是容易把我们引向线段树的错误方向。主要是这道题长得太容易让人想到线段树了：毕竟是区间问题。然而想到用线段树就想不出下一步了：如果用线段树，维护什么？怎么处理区间的合并？怎么快速得到答案？甚至可以问是对颜色建树还是对位置建树？不管怎么想，这道题都不好搞。

想到分块真的需要勇气，也或许是我的经验不足吧。毕竟要舍弃可能的$O(nlogn)$的做法，选择有时候容易被卡掉的$O(n\sqrt n)$的做法。实际上，即使用相对简单暴力的分块，也不那么好处理。

对位置分块。按照分块的套路，预处理出所有块的答案数组。这个可以用$O(n)$比较方便地实现。由于没有修改操作，接下来就直接考虑询问操作了。

询问操作的套路是，求出左端点、右端点所在块的编号，如果左右端点在同一块，就暴力扫一遍；如果不在同一块，就分别对中间经过的块、左端点到所在块的终点、右端点所在块的起点到右端点这三个部分进行处理，最后得到答案。然而这就有一个问题：如何得到中间经过的块合起来的答案？似乎不好处理，但是其实可以暴力。

假设已经知道了$i$到$j-1$块的答案$Ans$，现在讨论第$j$块。那么处理方法就是对第$j$块从左端点到右端点扫一遍，如果当前颜色在之前没有出现过，就不对$Ans$操作；否则如果当前颜色在之前出现过偶数次，就将$Ans$减1；如果当前颜色在之前出现过奇数次，就把$Ans$加1，做完上述的三个判断之后，更新当前颜色的出现次数。

为了实现上面的操作，可以看出我们要开一个$Cnt$数组记录每种颜色在当前讨论的范围内出现过多少次。同时也可以看出，上面的步骤可以用来预处理块与块两两之间的答案。预处理时间复杂度$O(n\sqrt n)$。

处理“左端点到所在块的终点、右端点所在块的起点到右端点”这两个部分也可以借鉴上面的思路。不过为了知道在某两块之间某种颜色出现了多少次，我们又需要开一个$Sum$数组记录某种颜色出现次数的前缀和。这个也可以预处理出来，时间复杂度$O(n)$。

通过这么多的预处理，终于把时间复杂度弄成了$O(n\sqrt n)$。在预处理时注意优化，在预处理时避免过多的memset。

更加具体的实现步骤参见代码。

---

#include<stdio.h>
#include<cmath>
#include<cstring>
#define MAXN 100005
#define MAXB 321

inline int _R()
{
    char s=getchar();int v=0;
    while(s>57||s<48)s=getchar();
    for(;s>47&&s<58;s=getchar())v=v*10+s-48;
    return v;
}

int Ans,A[MAXN],N,C,M,S,be[MAXN],v[MAXB][MAXB],Sum[MAXB][MAXN],cnt[MAXN];
int mark[MAXN],id;

int Tmp[MAXN];

void GetAns(int l,int r)
{
    Ans=0;

    int x=be[l],y=be[r],xx,yy,i,t;
    if(x==y)
    {
        for(i=l;i<=r;i++)
        {
            if(mark[A[i]]!=id)mark[A[i]]=id,Tmp[A[i]]=0;
            Tmp[A[i]]++;
            if((Tmp[A[i]]&1)&&Tmp[A[i]]!=1)Ans--;
            else if(!(Tmp[A[i]]&1))Ans++;
        }
        return;
    }
    Ans=v[x+1][y-1];
    xx=x*S;yy=(y-1)*S+1;
    for(i=l;i<=xx;i++)
    {
        if(mark[A[i]]!=id)mark[A[i]]=id,Tmp[A[i]]=0;
        Tmp[A[i]]++;
        t=Sum[y-1][A[i]]-Sum[x][A[i]]+Tmp[A[i]];
        if((t&1)&&t>1)Ans--;
        else if(!(t&1))Ans++;
    }
    for(i=yy;i<=r;i++)
    {
        if(mark[A[i]]!=id)mark[A[i]]=id,Tmp[A[i]]=0;
        Tmp[A[i]]++;
        t=Sum[y-1][A[i]]-Sum[x][A[i]]+Tmp[A[i]];
        if((t&1)&&t>1)Ans--;
        else if(!(t&1))Ans++;
    }
}

int main()
{
    int i,j,k,t,x,y;

    N=_R();C=_R();M=_R();
    for(i=1;i<=N;i++)A[i]=_R();

    S=sqrt(N);

    for(i=j=1;i<=N;i++)
    {
        be[i]=j;
        Sum[j][A[i]]++;
        if(i%S==0)j++;
    }

    memset(mark,-1,sizeof(mark));
    for(i=1;i<=be[N];i++)
    {
        for(j=i;j<=be[N];j++)
        {
            v[i][j]=v[i][j-1];
            x=(j-1)*S+1;y=j*S;
            for(k=x;k<=y;k++)
            {
                if(mark[A[k]]!=i-1)mark[A[k]]=i-1,cnt[A[k]]=1;
                else
                {
                    cnt[A[k]]^=1;
                    if(cnt[A[k]])v[i][j]--;
                    else v[i][j]++;
                }
            }
        }
    }

    for(i=1;i<=be[N];i++)
    for(j=1;j<=N;j++)Sum[i][j]+=Sum[i-1][j];

    memset(mark,0,sizeof(mark));

    for(i=1;i<=M;i++)
    {
        id++;
        scanf("%d%d",&x,&y);
        x=(Ans+x)%N+1;y=(Ans+y)%N+1;
        if(x>y)t=x,x=y,y=t;
        GetAns(x,y);
        printf("%d\n",Ans);
    }
}