题目链接:点击进入
题意
n 个数, t 次询问,每次询问给出一个区间边界 l , r ,对于每次询问输出 [ l , r ] 内的每个数字出现次数的平方与这个数乘积的(每个不同的数字计算一次,计算过的数字不再计算)。
例子:
思路
莫队模板题
代码
#pragma GCC optimize(2)
//#pragma GCC optimize(3)
#include<cstdio>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxn=1e6+10;
ll n,m,block,sum[maxn],a[maxn],ans[maxn],cnt;
struct node
{
int l;
int r;
int id;
int bl;
}p[maxn];
void add(int x)
{
cnt-=sum[a[x]]*sum[a[x]]*a[x];
sum[a[x]]++;
cnt+=sum[a[x]]*sum[a[x]]*a[x];
}
void del(int x)
{
cnt-=sum[a[x]]*sum[a[x]]*a[x];
sum[a[x]]--;
cnt+=sum[a[x]]*sum[a[x]]*a[x];
}
bool cmp(node x,node y)
{
if(x.bl!=y.bl) return x.l<y.l;
else
{
if(x.bl&1) return x.r<y.r;
else return x.r>y.r;
}
}
int main( )
{
// ios::sync_with_stdio(false);
scanf("%d%d",&n,&m);
block=sqrt(n);
for(register int i=1;i<=n;++i) cin>>a[i];
for(register int i=1;i<=m;++i)
{
int l,r;
scanf("%d%d",&l,&r);
p[i]=node{l,r,i,(l-1)/block+1};
}
sort(p+1,p+1+m,cmp);
int l=1,r=0;
for(register int i=1;i<=m;++i)
{
int ll=p[i].l,rr=p[i].r;
while(l<ll) del(l++);
while(l>ll) add(--l);
while(r<rr) add(++r);
while(r>rr) del(r--);
ans[p[i].id]=cnt;
}
for(register int i=1;i<=m;++i)
printf("%lld\n",ans[i]);
return 0;
}