bzoj3781 小B的询问

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

题解：
莫队
嗯..就莫队啊
按题目说的存个c[i]，根据$(c_i+1)^2=c_i^2+2*c_i+1$维护一下答案就好了。
注意顺序啊不要搞错了。虽然调一下就发现了。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long LL;
#define maxn 50010

struct node
{
    int l,r,id;
}q[maxn];
int qn,a[maxn];LL ans[maxn],c[maxn];
bool cmp(node x,node y)
{
    int xx=x.l/qn;
    int yy=y.l/qn;
    if (xx!=yy) return x.l<y.l;
    return x.r<y.r;
}
int main()
{
    //freopen("a.in","r",stdin);
    //freopen("a.out","w",stdout);
    int n,m,k,i,l,r;
    scanf("%d%d%d",&n,&m,&k);
    qn=(int)sqrt(n);
    for (i=1;i<=n;i++) scanf("%d",&a[i]);
    for (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);
    for (i=1;i<=k;i++) c[i]=0;

    LL sum=0;l=q[1].l;r=q[1].r;
    for (i=l;i<=r;i++)
    {
        sum+=2*c[a[i]]+1;
        c[a[i]]++;
    }
    ans[q[1].id]=sum;
    for (i=2;i<=m;i++)
    {
        while (l<q[i].l) {sum-=2*c[a[l]]-1;c[a[l]]--;l++;}
        while (l>q[i].l) {l--;sum+=2*c[a[l]]+1;c[a[l]]++;}
        while (r<q[i].r) {r++;sum+=2*c[a[r]]+1;c[a[r]]++;}
        while (r>q[i].r) {sum-=2*c[a[r]]-1;c[a[r]]--;r--;}
        ans[q[i].id]=sum;
    }

    for (i=1;i<=m;i++)
        printf("%lld\n",ans[i]);
    return 0;
}