Powerful array ( 莫队 )

最新推荐文章于 2023-03-29 23:41:56 发布

原创最新推荐文章于 2023-03-29 23:41:56 发布 · 265 阅读

0 ·

CC 4.0 BY-SA版权

真题练习同时被 2 个专栏收录

89 篇文章

订阅专栏

莫队

2 篇文章

订阅专栏

博客围绕一道题目展开，题目为给定 n 个数和 t 次询问，每次询问给出区间边界 l、r，需输出 [l, r] 内每个数字出现次数的平方与该数乘积（不同数字仅计算一次）。此题为莫队模板题，并给出了解题思路和代码。

题目链接：点击进入

题意

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