题意:
给你n个数,进行m个查询,每个查询有一个区间,如果1出现了cnt[1]次,2出现了cnt[2]次,3出现了cnt[3]次…..i出现cnt[i]次,那么就要求输出 1∗cnt[1]2+2∗cnt[2]2+3∗cnt[3]2+...+i∗cnt[i]2
思路:
莫队裸题,用cnt记录下某个数字出现,这样就能o(1)查询了(因为这个题时间要求高,map,vector会t),注意下long long,其他的就是莫队的正常操作了
错误及反思:
在没加inline以前,跑了4800ms+,加了以后3900ms,莫队因为每次转移区间都要调用函数,所以这样的优化极为明显
代码:
#include<bits/stdc++.h>
using namespace std;
const int maxn=200010;
int arr[maxn],pos[maxn],cnt[1000010];
int n,m;
long long now=0,ans[maxn];
struct query{
int l,r,id;
}Q[maxn];
bool cmp(query a,query b)
{
if(pos[a.l]==pos[b.l])
return a.r<b.r;
return a.l<b.l;
}
inline void update(int num,int p)
{
now-=1ll*cnt[num]*cnt[num]*num;
cnt[num]+=p;
now+=1ll*cnt[num]*cnt[num]*num;
}
int main(){
scanf("%d%d",&n,&m);
int block=sqrt(n);
for(int i=1;i<=n;i++)
{
scanf("%d",&arr[i]);
pos[i]=i/block;
}
for(int i=0;i<m;i++)
{
scanf("%d%d",&Q[i].l,&Q[i].r);
Q[i].id=i;
}
sort(Q,Q+m,cmp);
for(int i=0,l=1,r=0;i<m;i++)
{
for(;l<Q[i].l;l++)
update(arr[l],-1);
for(;l>Q[i].l;l--)
update(arr[l-1],1);
for(;r<Q[i].r;r++)
update(arr[r+1],1);
for(;r>Q[i].r;r--)
update(arr[r],-1);
ans[Q[i].id]=now;
}
for(int i=0;i<m;i++)
printf("%I64d\n",ans[i]);
}