题目链接:bzoj3781
题目大意:
小B有一个序列,包含N个1~K之间的整数。他一共有M个询问,每个询问给定一个区间[L..R],求Sigma(c(i)^2)的值,其中i的值从1到K,其中c(i)表示数字i在[L..R]中的重复次数。小B请你帮助他回答询问。
题解:
莫队
嗯..就莫队啊
按题目说的存个c[i],根据
(ci+1)2=c2i+2∗ci+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;
}