数据范围 2e5 ,可以带根号的算法,同时没有要求强制在线,于是可以考虑莫队。
我们把每次单点变化的贡献写出来就可以直接维护了。
代码:
#include<bits/stdc++.h>
using namespace std;
template <typename T>
inline void read(T &x){
x=0;char ch=getchar();bool f=false;
while(!isdigit(ch)){if(ch=='-'){f=true;}ch=getchar();}
while(isdigit(ch)){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}
x=f?-x:x;
return ;
}
template <typename T>
inline void write(T x){
if(x<0) putchar('-'),x=-x;
if(x>9) write(x/10);
putchar(x%10^48);
return ;
}
const int N=2e5+5,M=1e6+5;
#define ll long long
int n,m;
int bl[N];
ll cnt[M],a[N],Ans[N];
struct Query{int l,r,id;}Q[N];
ll Now;
inline void Add(int x){
Now+=2*cnt[a[x]]*a[x]+a[x];
cnt[a[x]]++;
return ;
}
inline void Del(int x){
Now+=-2*cnt[a[x]]*a[x]+a[x];
cnt[a[x]]--;
return ;
}
inline bool Cmp(Query x,Query y){return bl[x.l]^bl[y.l]?bl[x.l]<bl[y.l]:bl[x.l]&1?x.r<y.r:x.r>y.r;}
int main(){
read(n),read(m);
for(int i=1;i<=n;i++) read(a[i]);
const int t=sqrt(m);
for(int i=1;i<=m;i++){
read(Q[i].l),read(Q[i].r),Q[i].id=i;
bl[i]=(i-1)/t+1;
}
sort(Q+1,Q+m+1,Cmp);
int l=1,r=1;cnt[a[1]]++,Now=a[1];
for(int i=1;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]=Now;
}
for(int i=1;i<=m;i++) write(Ans[i]),putchar('
');
return 0;
}