CF86D Powerful array

题意：\(t\)次询问区间\([l,r]\)中\(\sum_x a_x\times cnt[a_x]^2\)，\(cnt\)为出现次数

裸的莫队题，用\(cnt[x]\)表示\(x\)出现的次数，根据平方的性质，我们得出加减操作

• 加：\(ans+=a_x\times 2cnt[a_x]+1,cnt[a_x]++\)
• 减：\(ans-=a_x\times 2cnt[a_x]-1,cnt[a_x]--\)

复杂度仍然是\(O(n\sqrt n)\)

Code

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#define N 2000000
using namespace std;
struct node
{
    int l,r,b,id;
}q[N+5];
int n,m,a[N+5],blo;
long long ans[N+5],tmp,cnt[N+5];
int cmp(node x,node y)     //奇偶性排序
{
    return (x.b==y.b)?(x.b%2==1?x.r<y.r:x.r>y.r):x.l<y.l;
}
void add(int x)
{
    tmp+=(long long)a[x]*1LL*(cnt[a[x]]*2+1);
    cnt[a[x]]++;
}
void del(int x)
{
    tmp-=(long long)a[x]*1LL*(cnt[a[x]]*2-1);
    cnt[a[x]]--;
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    blo=(sqrt(m*2/3))?n/sqrt(m*2/3):sqrt(n);  //块大小
    if (!blo)blo=sqrt(n);
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d",&q[i].l,&q[i].r);
        q[i].b=q[i].l/blo;
        q[i].id=i;
    }
    sort(q+1,q+m+1,cmp);
    for (int i=1,l=1,r=0;i<=m;i++)    //莫队
    {
        while (l>q[i].l)add(--l);
        while (r<q[i].r)add(++r);
        while (l<q[i].l)del(l++);
        while (r>q[i].r)del(r--);
        ans[q[i].id]=tmp;
    }
    for (int i=1;i<=m;i++)
        printf("%lld\n",ans[i]);
    return 0;
}