暑假集训Day14 I (莫队)

题目链接在这里：Problem - I - Codeforces

应该是一个比较经典的莫队题，一开始想的是这个数据范围肯定是要搞一个前缀和，后来发现如果弄前缀和的话区间还是不好操作，但是如果一位一位的算的话还是可以的，所以想到了莫队。

莫队要分块！！！不分块会T！

#include "bits/stdc++.h"
using namespace std;
typedef __int64 LL;
const int MAX=2e6+6;
LL n,t,a[MAX];
LL L,R,ans[MAX],cnt[MAX];
LL c[MAX];
struct Node{
    LL l,r,no;
}stu[MAX];
bool cmp(Node x,Node y){
    if (c[x.l]!=c[y.l]) return x.l<y.l;
    return x.r<y.r;
}
int main(){
    freopen ("i.in","r",stdin);
    freopen ("i.out","w",stdout);
    LL i,j,an;
    scanf("%I64d%I64d",&n,&t);
    for (i=1;i<=n;i++) scanf("%I64d",a+i);
    for (i=1,j=(LL)sqrt(n);i<=n;i++) c[i]=i/j;
    for (i=1;i<=t;i++){
        scanf("%I64d%I64d",&stu[i].l,&stu[i].r);
        stu[i].no=i;
    }
    sort(stu+1,stu+t+1,cmp);
    L=1,R=an=0;
    memset(cnt,0,sizeof(cnt));
    for (i=1;i<=t;i++){
        //cout<<stu[i].l<<' '<<stu[i].r<<endl;
        while (R<stu[i].r){
            R++;an+=(cnt[a[R]]*2+1)*a[R];
            cnt[a[R]]++;
        }//cout<<an<<" fuck"<<endl;
        while (L>stu[i].l){
            L--;an+=(cnt[a[L]]*2+1)*a[L];
            cnt[a[L]]++;
        }
        while (R>stu[i].r){
            an-=((cnt[a[R]]-1)*2+1)*a[R];
            cnt[a[R]]--;
            R--;
        }
        while (L<stu[i].l){
            an-=((cnt[a[L]]-1)*2+1)*a[L];
            cnt[a[L]]--;
            L++;
        }
        ans[stu[i].no]=an;
    }
    for (i=1;i<=t;i++) printf("%I64d
",ans[i]);
    return 0;
}