BZOJ3781小B的询问

3781: 小B的询问
Time Limit: 10 Sec Memory Limit: 128 MB
Submit: 417 Solved: 278
Description
小B有一个序列，包含N个1~K之间的整数。他一共有M个询问，每个询问给定一个区间[L..R]，求Sigma(c(i)^2)的值,其中i的值从1到K，其中c(i)表示数字i在[L..R]中的重复次数。小B请你帮助他回答询问。
Input
第一行，三个整数N、M、K。
第二行，N个整数，表示小B的序列。
接下来的M行，每行两个整数L、R。
Output
M行，每行一个整数，其中第i行的整数表示第i个询问的答案。
Sample Input
6 4 3
1 3 2 1 1 3
1 4
2 6
3 5
5 6
Sample Output
6
9
5
2
HINT
对于全部的数据，1<=N、M、K<=50000
普通莫队算法。。
题不能再裸了。。人不能再弱了。。
附上本蒟蒻的代码：

#include<cstdio>#include<cmath>#include<algorithm>using namespace std;int n,m,k,b[50001];long long cnt[50001],res[50001];struct kx{    int l,r,block,id;}a[50001];bool operator < (kx a,kx b){    if (a.block==b.block)      return a.r<b.r;    return a.block<b.block;}int read(){    int w=0,c=1;    char ch=getchar();    while (ch<'0' || ch>'9')      {        if (ch=='-')          c=-1;        ch=getchar();      }    while (ch>='0' && ch<='9')      {        w=w*10+ch-'0';        ch=getchar();      }    return w*c;}int main(){    int size,l=1,r=0,i;    long long ans=0;    n=read(),m=read(),k=read();    size=sqrt(n);    for (i=1;i<=n;i++)      b[i]=read();    for (i=1;i<=m;i++)      {        a[i].l=read(),a[i].r=read();        a[i].id=i,a[i].block=(a[i].l-1)/size+1;      }    sort(a+1,a+m+1);    for (i=1;i<=m;i++)      {        while (l>a[i].l)          l--,cnt[b[l]]++,ans+=2*cnt[b[l]]-1;        while (r<a[i].r)          r++,cnt[b[r]]++,ans+=2*cnt[b[r]]-1;        while (l<a[i].l)          cnt[b[l]]--,ans-=2*cnt[b[l]]+1,l++;        while (r>a[i].r)          cnt[b[r]]--,ans-=2*cnt[b[r]]+1,r--;        res[a[i].id]=ans;      }    for (i=1;i<=m;i++)      printf("%lld\n",res[i]);    return 0;}