BZOJ 3781小B的询问

题意

小B有一个序列，包含N个1~K之间的整数。他一共有M个询问，每个询问给定一个区间[L..R]，求Sigma(c(i)^2)的值,其中i的值从1到K，其中c(i)表示数字i在[L..R]中的重复次数。小B请你帮助他回答询问。

解

裸莫队

代码

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#define For(i,j,k) for(int i=(j);i<=(int)k;i++)
#define Forr(i,j,k) for(int i=(j);i>=(int)k;i--)
#define Set(a,b) memset(a,b,sizeof(a))
#define ll long long 
#define sqr(x) ((x)*(x))
using namespace std;
const int N=50010;
template <class T>inline void read(T &x){
    x=0;char c=getchar();
    while(c<'0'||c>'9')c=getchar();
    while(c>='0'&&c<='9')x=(x<<1)+(x<<3)+(c^48),c=getchar();
}
int n,m,k;
int c[N],ans,p[N],blo,v[N];
struct node{
    int l,r,id;
    ll s;
}a[N];
inline bool cmpb(node a,node b){
    return p[a.l]<p[b.l]||(p[a.l]==p[b.l]&&a.r<b.r);
}
inline bool cmpid(node a,node b){
    return a.id<b.id;
}
inline void init(){
    read(n),read(m),read(k);
    For(i,1,n)read(v[i]);
    For(i,1,m)read(a[i].l),read(a[i].r),a[i].id=i;
    blo=(int)sqrt(n);
    For(i,1,n)p[i]=(i-1)/blo+1;
}
inline int update(int i,int add){
    ans-=sqr(c[v[i]]),c[v[i]]+=add,ans+=sqr(c[v[i]]);
}
inline void solve(){
    sort(a+1,a+m+1,cmpb);
    for(int i=1,l=1,r=0,L,R;i<=m;i++){
        L=a[i].l,R=a[i].r;
        while(r<R)update(r+1,1),r++;
        while(r>R)update(r,-1),r--;
        while(l<L)update(l,-1),l++;
        while(l>L)update(l-1,1),l--;
        a[i].s=ans;
    }
    sort(a+1,a+m+1,cmpid);
    For(i,1,m)printf("%lld\n",a[i].s);
}
int main(){
    init();
    solve();
    return 0;
}