【莫队算法】bzoj3781 小B的询问

莫队经典。 开个数组维护a[i]出现的次数。

 #include<cstdio>
 #include<cmath>
 #include<algorithm>
 using namespace std;
 int Num,CH[],f,c;
 inline void R(int &x){
     c=;f=;
     for(;c<''||c>'';c=getchar())if(c=='-')f=-;
     for(x=;c>=''&&c<='';c=getchar())(x*=)+=(c-'');
     x*=f;
 }
 inline void P(int x){
     if(x<)putchar(x+'');
     else{P(x/);putchar(x%+'');}
 }
 int ans,anss[];
 int n,m,K,num[],a[],T[];
 int sqr(const int &x){return x*x;}
 struct ASK{int l,r,p;void Read(){R(l);R(r);}}Q[];
 bool operator < (const ASK &a,const ASK &b){return num[a.l]!=num[b.l] ? num[a.l]<num[b.l] : a.r<b.r;}
 void makeblock()
 {
     int sz=sqrt(n),sum=; if(!sz) sz=;
     for(;sum*sz<n;++sum)
       {
           int r=sum*sz;
           for(int i=sz*(sum-)+;i<=r;++i) num[i]=sum;
       }
     for(int i=sz*(sum-)+;i<=n;++i) num[i]=sum;
 }
 int main()
 {
     R(n);R(m);R(K);makeblock();
     for(int i=;i<=n;++i) R(a[i]);
     for(int i=;i<=m;++i) {Q[i].Read(); Q[i].p=i;}
     sort(Q+,Q+m+);
     for(int i=Q[].l;i<=Q[].r;++i){ans-=sqr(T[a[i]]);ans+=sqr(++T[a[i]]);}
     anss[Q[].p]=ans;
     for(int i=;i<=m;++i)
       {
           if(Q[i].l<Q[i-].l){for(int j=Q[i-].l-;j>=Q[i].l;--j){ans-=sqr(T[a[j]]);ans+=sqr(++T[a[j]]);}}
           else{for(int j=Q[i-].l;j<Q[i].l;++j){ans-=sqr(T[a[j]]);ans+=sqr(--T[a[j]]);}}
           if(Q[i].r<Q[i-].r){for(int j=Q[i-].r;j>Q[i].r;--j){ans-=sqr(T[a[j]]);ans+=sqr(--T[a[j]]);}}
           else{for(int j=Q[i-].r+;j<=Q[i].r;++j){ans-=sqr(T[a[j]]);ans+=sqr(++T[a[j]]);}}
         anss[Q[i].p]=ans;
       } for(int i=;i<=m;++i) P(anss[i]),puts("");
     return ;
 }