【题解】Luogu CF86D Powerful array

原题传送门

裸的莫队啊，我博客里有对莫队较详细的介绍

这道题很简单，可以说是裸的模板

但是如何在已有的值上进行操作？

小学生应该都知道

那么转移就超级简单了qaq

inline void add(register int c)
{
    sum+=(num[c]+num[c]+1)*c;
    ++num[c];
}
inline void del(register int c)
{
    --num[c];
    sum-=(num[c]+num[c]+1)*c;
}

要开long long（虽说我没试过int，但我觉得会炸int）

#include <bits/stdc++.h>
#define N 200005
using namespace std;
inline int read()
{
    register int x=0,f=1;register char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
    return x*f;
}
inline void write(register long long x)
{
    if(!x)putchar('0');if(x<0)x=-x,putchar('-');
    static int sta[25];int tot=0;
    while(x)sta[tot++]=x%10,x/=10;
    while(tot)putchar(sta[--tot]+48);
}
struct query{
    int l,r,id,bl;
}q[N];
int a[N],blocksize=0,num[1000005];
long long ans[N],sum=0;
inline void add(register int c)
{
    sum+=(num[c]+num[c]+1)*c;
    ++num[c];
}
inline void del(register int c)
{
    --num[c];
    sum-=(num[c]+num[c]+1)*c;
}
inline bool cmp(register query a,register query b)
{
    return a.bl!=b.bl?a.l<b.l:(a.bl&1?a.r<b.r:a.r>b.r);
}
int main()
{
    int n=read(),m=read();
    blocksize=sqrt(n);
    for(register int i=1;i<=n;++i)
        a[i]=read();
    for(register int i=1;i<=m;++i)
    {
        int l=read(),r=read();
        q[i]=(query){l,r,i,(l-1)/blocksize+1};
    }
    sort(q+1,q+1+m,cmp);
    int l=1,r=0;
    for(register int i=1;i<=m;++i)
    {
        int ll=q[i].l,rr=q[i].r;
        while(ll<l)
            add(a[--l]);
        while(ll>l)
            del(a[l++]);
        while(rr>r)
            add(a[++r]);
        while(rr<r)
            del(a[r--]);
        ans[q[i].id]=sum;
    }
    for(register int i=1;i<=m;++i)
        write(ans[i]),printf("\n");
    return 0;
 } 

转载于:https://www.cnblogs.com/yzhang-rp-inf/p/10009499.html