bzoj2821作诗

http://www.lydsy.com/JudgeOnline/problem.php?id=2821

分块

我们把数列分成$\sqrt{N}$块

记$f[i][j]$表示第i块到第j块的答案，这个可以在$O(N\sqrt{N})$内得到。

记$g[i][j]$第1到第i块中数字j出现了多少次，这个我们可以先求出第i块中数字j出现了多少次，然后求前缀和即可，这个可以在$O(C\sqrt{N})$内得到。

对于询问区间[l,r]我们可以从f数组中快速求出中间连续的完整的块答案。

对于剩余部分，我们可以一个一个调整答案，反正剩余部分的长度是$\sqrt{N}$级别的。

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<fstream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<utility>
#include<set>
#include<bitset>
#include<vector>
#include<functional>
#include<deque>
#include<cctype>
#include<climits>
#include<complex>
//#include<bits/stdc++.h>适用于CF,UOJ，但不适用于poj

using namespace std;

typedef long long LL;
typedef double DB;
typedef pair<int,int> PII;
typedef complex<DB> CP;

#define mmst(a,v) memset(a,v,sizeof(a))
#define mmcy(a,b) memcpy(a,b,sizeof(a))
#define fill(a,l,r,v) fill(a+l,a+r+1,v)
#define re(i,a,b)  for(i=(a);i<=(b);i++)
#define red(i,a,b) for(i=(a);i>=(b);i--)
#define ire(i,x) for(typedef(x.begin()) i=x.begin();i!=x.end();i++)
#define fi first
#define se second
#define m_p(a,b) make_pair(a,b)
#define p_b(a) push_back(a)
#define SF scanf
#define PF printf
#define two(k) (1<<(k))

template<class T>inline T sqr(T x){return x*x;}
template<class T>inline void upmin(T &t,T tmp){if(t>tmp)t=tmp;}
template<class T>inline void upmax(T &t,T tmp){if(t<tmp)t=tmp;}

const DB EPS=1e-;
inline int sgn(DB x){if(abs(x)<EPS)return ;return(x>)?:-;}
const DB Pi=acos(-1.0);

inline int gint()
  {
        int res=;bool neg=;char z;
        for(z=getchar();z!=EOF && z!='-' && !isdigit(z);z=getchar());
        if(z==EOF)return ;
        if(z=='-'){neg=;z=getchar();}
        for(;z!=EOF && isdigit(z);res=res*+z-'',z=getchar());
        return (neg)?-res:res;
    }
inline LL gll()
  {
      LL res=;bool neg=;char z;
        for(z=getchar();z!=EOF && z!='-' && !isdigit(z);z=getchar());
        if(z==EOF)return ;
        if(z=='-'){neg=;z=getchar();}
        for(;z!=EOF && isdigit(z);res=res*+z-'',z=getchar());
        return (neg)?-res:res;
    }

const int maxN=;
const int maxC=;
const int maxcnt=;

int N,C,Q;
int a[maxN+];
int cnt,len,l[maxcnt+],r[maxcnt+];
int id[maxN+];
int f[maxcnt+][maxcnt+];
int g[maxcnt+][maxC+];
int t[maxC+];
int ans;

int main()
  {
      freopen("bzoj2821.in","r",stdin);
      freopen("bzoj2821.out","w",stdout);
      int i,j,k;
      N=gint();C=gint();Q=gint();
        re(i,,N)a[i]=gint();
      len=int(sqrt(DB(N)));
      re(i,,N)
        {
            if((i-)%len==)r[cnt]=i-,l[++cnt]=i;
            id[i]=cnt;
        }
      r[cnt]=N;
      re(i,,cnt)
        {
            int res=;
            re(j,i,cnt)
              {
                re(k,l[j],r[j])
                          {
                              t[a[k]]++;
                              if(t[a[k]]>= && !(t[a[k]]&))res++;
                              if(t[a[k]]>= &&  (t[a[k]]&))res--;
                            }
                        f[i][j]=res;
                    }
                re(k,l[i],N)t[a[k]]--;
            }
        re(i,,cnt)re(j,l[i],r[i])g[i][a[j]]++;
        re(i,,cnt)re(j,,C)g[i][j]+=g[i-][j];
        ans=;
        while(Q--)
          {
              int L=(gint()+ans)%N+,R=(gint()+ans)%N+,res=;
              if(L>R)swap(L,R);
              if(id[L]==id[R] || id[L]+==id[R])
                {
                    re(i,L,R)
                      {
                          t[a[i]]++;
                          if(t[a[i]]>= && !(t[a[i]]&))res++;
                          if(t[a[i]]>= &&  (t[a[i]]&))res--;
                      }
                    re(i,L,R)t[a[i]]--;
                    ans=res;
                }
              else
                {
                    int p=(L==l[id[L]])?id[L]:id[L]+,q=(R==r[id[R]])?id[R]:id[R]-;
                    res=f[p][q];
                    red(i,l[p]-,L)
                      {
                          t[a[i]]++;
                          t[a[i]]+=g[q][a[i]]-g[p-][a[i]];
                          if(t[a[i]]>= && !(t[a[i]]&))res++;
                          if(t[a[i]]>= &&  (t[a[i]]&))res--;
                          t[a[i]]-=g[q][a[i]]-g[p-][a[i]];
                      }
                    re(i,r[q]+,R)
                      {
                          t[a[i]]++;
                          t[a[i]]+=g[q][a[i]]-g[p-][a[i]];
                          if(t[a[i]]>= && !(t[a[i]]&))res++;
                          if(t[a[i]]>= &&  (t[a[i]]&))res--;
                          t[a[i]]-=g[q][a[i]]-g[p-][a[i]];
                      }
                    red(i,l[p]-,L)t[a[i]]--;
                    re(i,r[q]+,R)t[a[i]]--;
                    ans=res;
                }
              PF("%d\n",ans);
          }
        return ;
    }