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(nn−−√) 的做法。实际上,即使用相对简单暴力的分块,也不那么好处理。
对位置分块。按照分块的套路,预处理出所有块的答案数组。这个可以用 O(n) 比较方便地实现。由于没有修改操作,接下来就直接考虑询问操作了。
询问操作的套路是,求出左端点、右端点所在块的编号,如果左右端点在同一块,就暴力扫一遍;如果不在同一块,就分别对中间经过的块、左端点到所在块的终点、右端点所在块的起点到右端点这三个部分进行处理,最后得到答案。然而这就有一个问题:如何得到中间经过的块合起来的答案?似乎不好处理,但是其实可以暴力。
假设已经知道了
i
到
j−1
块的答案
Ans
,现在讨论第
j
块。那么处理方法就是对第
为了实现上面的操作,可以看出我们要开一个 Cnt 数组记录每种颜色在当前讨论的范围内出现过多少次。同时也可以看出,上面的步骤可以用来预处理块与块两两之间的答案。预处理时间复杂度 O(nn−−√) 。
处理“左端点到所在块的终点、右端点所在块的起点到右端点”这两个部分也可以借鉴上面的思路。不过为了知道在某两块之间某种颜色出现了多少次,我们又需要开一个 Sum 数组记录某种颜色出现次数的前缀和。这个也可以预处理出来,时间复杂度 O(n) 。
通过这么多的预处理,终于把时间复杂度弄成了 O(nn−−√) 。在预处理时注意优化,在预处理时避免过多的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);
}
}