题意:Ks为区间内s的数目,求区间[L,R]之间所有Ks*Ks*s的和
分析:用莫队算法可以做,时间卡得有点紧,写的时候注意
#include <iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#define LL long long
#define EPS 1e-9
#include<cmath>
#define N 200005
#define M 1000005
using namespace std;
LL ans[N],tmp;
int a[N];
LL num[M];
struct query{
int l,r,id,pos;
bool operator <(const query &k)const{
return pos!=k.pos?pos<k.pos:r<k.r;
}
}q[N];
void insert(int x){
tmp+=((num[a[x]]<<1)+1)*a[x]; //这里容易超时
num[a[x]]++;
}
void erase(int x){
num[a[x]]--;
tmp-=((num[a[x]]<<1)+1)*a[x];
}
int main()
{
int n,t,i,l,r,unit;
while(~scanf("%d%d",&n,&t)){
memset(num,0,sizeof(num));
for(i=1;i<=n;i++){
scanf("%d",&a[i]);
}
unit=sqrt((double)n);
for(i=0;i<t;i++){
scanf("%d%d",&q[i].l,&q[i].r);
q[i].id=i;
q[i].pos=q[i].l/unit;
}
sort(q,q+t);
l=1;
r=0;
tmp=0;
for(i=0;i<t;i++){
while(r<q[i].r)insert(++r);
while(r>q[i].r)erase(r--);
while(l<q[i].l)erase(l++);
while(l>q[i].l)insert(--l);
ans[q[i].id]=tmp;
}
for(i=0;i<t;i++){
printf("%I64d\n",ans[i]);
}
}
return 0;
}