Powerful array ( 莫队 )

题目链接：点击进入

题意

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;
}