BZOJ 3781 莫队算法

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<ctime>
#include<cstring>
#include<string>
#include<iostream>
#include<iomanip>
#include<algorithm>
using namespace std;
struct query
{
int l,r,pos;
bool operator <(query b) const
{
if((l>>8) != (b.l>>8)) return l<b.l;
return r<b.r;
}
}q[50050];
long long sum[50050];
long long now;
long long ans[50050];
int a[50050];
void update(int x,int y)
{
now-=sum[x]*sum[x];
sum[x]+=y;
now+=sum[x]*sum[x];
}
int main()
{
int n,m,k;
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=1;i<=m;i++) scanf("%d%d",&q[i].l,&q[i].r),q[i].pos=i;
sort(q+1,q+1+m);
int l=1,r=0;
for(int i=1;i<=m;i++)
{
while(r<q[i].r) update(a[++r],1);
while(r>q[i].r) update(a[r--],-1);
while(l<q[i].l) update(a[l++],-1);
while(l>q[i].l) update(a[--l],1);
ans[q[i].pos]=now;
}
for(int i=1;i<=m;i++)
printf("%lld\n",ans[i]);
return 0;
}