Powerful array（莫队奇偶优化）

Posted 2021-11-09 thusloop

Powerful array

//#pragma GCC optimize(2)
//#pragma GCC optimize(3,"Ofast","inline")
#include<bits/stdc++.h>
#define int long long
#define fi first
#define se second
#define pb push_back
#define pii pair<int,int>
#define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
const int inf=(int)8e18+100;
const int maxn=1e6+100;
int a[maxn],pos[maxn],ans[maxn],sum,cnt[maxn];
struct node
{
	int l,r;
	int id;
} t[maxn];
bool cmp(node a,node b)
{
	if(pos[a.l]!=pos[b.l]) return a.l<b.l;	
	if(pos[a.l]&1) return a.r<b.r;
	return a.r>b.r;	
}
void add(int x)
{
	sum-=cnt[a[x]]*cnt[a[x]]*a[x];
	cnt[a[x]]++;
	sum+=cnt[a[x]]*cnt[a[x]]*a[x];
}
void del(int x)
{
	sum-=cnt[a[x]]*cnt[a[x]]*a[x];
	cnt[a[x]]--;
	sum+=cnt[a[x]]*cnt[a[x]]*a[x];
}
signed main()
{
	IOS
	int n,q,sz;
	cin>>n>>q;
	sz=sqrt(n);
	for(int i=1; i<=n; i++)
	{
		cin>>a[i];
		pos[i]=i/sz;
	}
	for(int i=1; i<=q; i++)
	{
		int l,r;
		cin>>l>>r;
		t[i]= {l,r,i};
	}
	sort(t+1,t+q+1,cmp);
	int l=1,r=0;
	for(int i=1; i<=q; i++)
	{
		while(l>t[i].l)add(--l);
		while(r<t[i].r)add(++r);
		while(l<t[i].l)del(l++);
		while(r>t[i].r)del(r--);
		ans[t[i].id]=sum;
	}
	for(int i=1; i<=q; i++)
	{
		cout<<ans[i]<<"\\n";
	}
}