3781: 小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
【解题报告】
莫队裸题。
讲道理现在发现莫队其实没什么难的。
虽然证不来它的复杂度
代码如下:
/**************************************************************
Problem: 3781
User: onepointo
Language: C++
Result: Accepted
Time:936 ms
Memory:2200 kb
****************************************************************/
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
#define N 50010
int block,cnt[N],pos[N],ans[N];
int n,m,k,num[N],tmp;
struct Data
{
int l,r,id;
friend bool operator < (const Data &a,const Data &b)
{return pos[a.l]==pos[b.l]?a.r<b.r:a.l<b.l;}
}a[N];
void update(int p,int add)
{
tmp-=cnt[p]*cnt[p];
cnt[p]+=add;
tmp+=cnt[p]*cnt[p];
}
void solve()
{
for(int i=1,l=1,r=0;i<=m;i++)
{
while(r<a[i].r) update(num[++r],1);
while(l>a[i].l) update(num[--l],1);
while(r>a[i].r) update(num[r--],-1);
while(l<a[i].l) update(num[l++],-1);
ans[a[i].id]=tmp;
}
}
int main()
{
scanf("%d%d%d",&n,&m,&k);
block=(int)sqrt(n);
for(int i=1;i<=n;++i) scanf("%d",&num[i]);
for(int i=1;i<=m;++i)
{
a[i].id=i;
scanf("%d%d",&a[i].l,&a[i].r);
pos[a[i].l]=(a[i].l-1)/block+1;
}
sort(a+1,a+m+1);
solve();
for(int i=1;i<=m;++i) printf("%d\n",ans[i]);
return 0;
}