Powerful array（莫队奇偶优化）

最新推荐文章于 2024-01-25 11:32:19 发布

thusloop

最新推荐文章于 2024-01-25 11:32:19 发布

阅读量199

点赞数

分类专栏：莫队文章标签： c++ 开发语言后端

本文链接：https://blog.csdn.net/qq_51737737/article/details/120890370

版权

莫队专栏收录该内容

7 篇文章 0 订阅

订阅专栏

这篇博客探讨了一种高效处理数组并进行区间查询的方法。通过预处理数组，使用sqrt分解和排序，实现了O(log n)复杂度的区间更新和查询。代码中展示了如何维护每个元素出现次数的平方和，用于解决特定查询问题。

摘要由CSDN通过智能技术生成

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