【題解】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;
 }
相關文章
相關標籤/搜索