BZOJ3781 小B的询问

原题链接：https://www.lydsy.com/JudgeOnline/problem.php?id=3781

小B的询问

Description

小B有一个序列，包含N个1~K之间的整数。他一共有M个询问，每个询问给定一个区间[L..R]，求$\sum_{i=L}^Rc(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<bits/stdc++.h>
#define A(v) cot[v]*cot[v]
using namespace std;
const int M=5e4+5;
struct sd{int le,ri,id;};
sd ask[M];
int n,m,k,cot[M],val[M],ans[M],SZ,res;
bool operator <(sd a,sd b){return a.le/SZ==b.le/SZ?a.ri/SZ<b.ri/SZ:a.le/SZ<b.le/SZ;}
void in(){scanf("%d%d%d",&n,&m,&k);SZ=sqrt(n);for(int i=1;i<=n;++i)scanf("%d",&val[i]);}
void add(int v){res-=A(v);cot[v]++;res+=A(v);}
void del(int v){res-=A(v);cot[v]--;res+=A(v);}
void ac()
{
    int a,b,L,R;
    for(int i=1;i<=m;++i){scanf("%d%d",&ask[i].le,&ask[i].ri);ask[i].id=i;}
    sort(ask+1,ask+1+m);
    L=R=ask[1].le,cot[val[ask[1].le]]=1,res=1;
    for(int i=1;i<=m;++i)
    {
        while(L<ask[i].le)del(val[L++]);
        while(L>ask[i].le)add(val[--L]);
        while(R>ask[i].ri)del(val[R--]);
        while(R<ask[i].ri)add(val[++R]);
        ans[ask[i].id]=res;
    }
    for(int i=1;i<=m;++i)printf("%d\n",ans[i]);
}
int main()
{
    in();ac();
    return 0;
}