Yandex.Algorithm 2011 Round 2 D - Powerful array (莫队算法)

题意：

给你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]);
}