BZOJ3781: 小B的询问

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

这道题是权限题，先不搞传送门了~~

实在是手痒想练一练新学的莫队算法，于是借了同学一个权限号打了几波水题
这是第一道1A的题目，很感动（可能是因为太水了的缘故吧~~~~）
裸的莫队，连公式都不带，进去change的时候判断一下它是不是在K以内就好

代码如下：

#include<cmath>#include<cstdio>#include<cstring>#include<cstdlib>#include<algorithm> using namespace std;typedef long long ll;int n,m,unit,Be[510000],k;struct  Mo{    int l,r,id;    ll x;}q[510000];int a[510000],c[510000];ll ans;bool cmp(Mo a,Mo b){return Be[a.l]==Be[b.l]?a.r<b.r:a.l<b.l;}bool Cmp(Mo a,Mo b){return a.id<b.id;}ll S(int x){return x*x;}void change(int x,int d){    if(a[x]>k)return ;    ans-=S(c[a[x]]);c[a[x]]+=d;ans+=S(c[a[x]]);}int main(){    scanf("%d%d%d",&n,&m,&k);ans=0,unit=sqrt(n);    for(int i=1;i<=n;i++)scanf("%d",&a[i]),Be[i]=i/unit+1;    for(int i=1;i<=m;i++)scanf("%d%d",&q[i].l,&q[i].r),q[i].id=i;    sort(q+1,q+1+m,cmp);    int l=1,r=0;    for(int i=1;i<=m;i++)    {        while(l<q[i].l)change(l,-1),l++;        while(l>q[i].l)change(l-1,1),l--;        while(r<q[i].r)change(r+1,1),r++;        while(r>q[i].r)change(r,-1),r--;        q[i].x=ans;    }    sort(q+1,q+1+m,Cmp);    for(int i=1;i<=m;i++)printf("%lld\n",q[i].x);    return 0;}

by_lmy