codeforces 86 D Powerful array

题目链接

题目大意

给定一个有n个元素的序列，有m个询问。

定义Si是i在数组中出现的次数。

定义一个数组的价值是∑Si∗Si∗i.

每次询问[L,R]的价值。

题解

似乎就是莫队的模板题啊。

记录每个元素出现的次数cnti

每次加入一个元素时，答案就加上cnti+12−cnti2。

每次删除一个元素时，答案就减去cnti2−cnti−12。

好了。

这题貌似有点卡常，我一开始为了方便，把cnt开成long long，结果就TLE了。

还是要在乘法时强制转化。

#include<cmath>
#include<cstdio>
#include<algorithm>
using namespace std;
const int M=200005;
typedef long long ll;
struct Query{
int L,R,id;
}q[M];
int s,col[M];
ll ans[M];
int cnt[M*10];
bool cmp(Query a,Query b){
if(a.L/s==b.L/s) return a.R<b.R;
return a.L/s<b.L/s;
}
inline void rd(int&res){
res=0;char c;
while(c=getchar(),c<48);
do res=res*10+(c^48);
while(c=getchar(),c>47);
}
void print(ll x){
if(!x) return;
print(x/10);
putchar(x%10^48);
}
inline void sc(ll x){
if(x==0) putchar('0');
else print(x);
putchar('\n');
}
int main(){
int n,m;
scanf("%d%d",&n,&m);
s=(int)sqrt(n);
for(int i=1;i<=n;i++)
rd(col[i]);
for(int i=0;i<m;i++){
int a,b;
rd(a);rd(b);
q[i]=(Query){a,b,i};
}
sort(q,q+m,cmp);
int L=1,R=0;
ll res=0;
for(int i=0;i<m;i++){
while(R<q[i].R){
R++;
res+=(ll)col[R]*(cnt[col[R]]+1)*(cnt[col[R]]+1)-(ll)col[R]*cnt[col[R]]*cnt[col[R]];
cnt[col[R]]++;
}
while(L<q[i].L){
res-=(ll)col[L]*cnt[col[L]]*cnt[col[L]]-(ll)col[L]*(cnt[col[L]]-1)*(cnt[col[L]]-1);
cnt[col[L]]--;
L++;
}
while(R>q[i].R){
res-=(ll)col[R]*cnt[col[R]]*cnt[col[R]]-(ll)col[R]*(cnt[col[R]]-1)*(cnt[col[R]]-1);
cnt[col[R]]--;
R--;
}
while(L>q[i].L){
L--;
res+=(ll)col[L]*(cnt[col[L]]+1)*(cnt[col[L]]+1)-(ll)col[L]*cnt[col[L]]*cnt[col[L]];
cnt[col[L]]++;
}
ans[q[i].id]=res;
}
for(int i=0;i<m;i++)
sc(ans[i]);
return 0;
}