原题链接:https://www.lydsy.com/JudgeOnline/problem.php?id=3781
小B的询问
Description
小B有一个序列,包含N个1~K之间的整数。他一共有M个询问,每个询问给定一个区间[L..R],求 ∑Ri=Lc(i)2 ∑ i = L R 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<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;
}