P2709 小B的询问 ( 莫队 )

最新推荐文章于 2025-06-15 14:33:54 发布

原创最新推荐文章于 2025-06-15 14:33:54 发布 · 230 阅读

0 ·

CC 4.0 BY-SA版权

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

89 篇文章

订阅专栏

莫队

2 篇文章

订阅专栏

博客围绕莫队经典例题展开，给出正常版代码和卡常 + O2 优化版代码，还提醒定义 int 时不要输入 lld 以免出错，强调莫队卡常 + O2 速度快。

题目描述：点击进入

在这里插入图片描述

思路

莫队经典例题
保存个模板吧
哦，对了，每日一错：定义的 int 不要输入 lld 要不 WA 到你怀疑人生/(ㄒoㄒ)/~~
(顺便说一句，莫队卡常 + O2 真快)
如图
正常版
在这里插入图片描述
卡常 + O2

代码1（正常）

//#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=1e5+10;
int n,m,k,block,cnt,sum[maxn],a[maxn],ans[maxn];
struct node
{
	int l;
	int r;
	int id;
	int bl;
}p[maxn];
void add(int x)
{
	cnt-=sum[a[x]]*sum[a[x]];
	sum[a[x]]++;
	cnt+=sum[a[x]]*sum[a[x]];
}
void del(int x)
{
	cnt-=sum[a[x]]*sum[a[x]];
	sum[a[x]]--;
	cnt+=sum[a[x]]*sum[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%d",&n,&m,&k);
	block=sqrt(n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(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(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(int i=1;i<=m;i++) 
	    printf("%d\n",ans[i]);
	return 0;
}

代码2（卡常 + O2 ）

//#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=1e5+10;
int n,m,k,block,cnt,sum[maxn],a[maxn],ans[maxn];
struct node
{
	int l;
	int r;
	int id;
	int bl;
}p[maxn];
void add(int x)
{
	cnt-=sum[a[x]]*sum[a[x]];
	sum[a[x]]++;
	cnt+=sum[a[x]]*sum[a[x]];
}
void del(int x)
{
	cnt-=sum[a[x]]*sum[a[x]];
	sum[a[x]]--;
	cnt+=sum[a[x]]*sum[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%d",&n,&m,&k);
	block=sqrt(n);
	for(register int i=1;i<=n;++i) scanf("%d",&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("%d\n",ans[i]);
	return 0;
}