【bzoj3781】小B的询问莫队_莫队题目-优快云博客

本文详细解析了一道经典的莫队算法题目，通过简洁的代码实现和详细的步骤说明，帮助读者理解如何利用莫队算法解决区间查询问题。文章还介绍了算法的具体实现细节，包括数据结构的设计和优化技巧。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

AC通道：http://www.lydsy.com/JudgeOnline/problem.php?id=3781

【题解】

很裸的莫队题目，好久没有做这么水的题了。

每次左右端点移动，答案先减去原贡献，再加上现在的贡献就行了。

/*************
  bzoj 3781
  by chty
  2016.11.17
*************/
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<ctime>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;
#define FILE "read"
#define MAXN 50010
#define up(i,j,n) for(ll i=j;i<=n;i++)
struct node{ll x,y,id;}q[MAXN];
ll n,m,k,now,block,a[MAXN],ans[MAXN],cnt[MAXN];
namespace Init{
	char buf[1<<15],*fs,*ft;
	inline char getc() {return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;}
	inline ll read(){
		ll x=0,f=1;  char ch=getc();
		while(!isdigit(ch))  {if(ch=='-')  f=-1;  ch=getc();}
		while(isdigit(ch))  {x=x*10+ch-'0';  ch=getc();}
		return x*f;
	}
}using namespace Init;
namespace Mo_Team{
	bool cmp(node a,node b) {return a.x/block==b.x/block?a.y<b.y:a.x/block<b.x/block;}
	void updata(ll x,ll v){now-=cnt[x]*cnt[x];cnt[x]+=v;now+=cnt[x]*cnt[x];}
	void work(){
		sort(q+1,q+m+1,cmp);
		ll l(1),r(0);
		up(i,1,m){
			while(r<q[i].y)  updata(a[++r],1);
			while(r>q[i].y)  updata(a[r--],-1);
			while(l<q[i].x)  updata(a[l++],-1);
			while(l>q[i].x)  updata(a[--l],1);
			ans[q[i].id]=now;
		}
		up(i,1,m)  printf("%lld\n",ans[i]);
	}
}
void init(){
	n=read();  m=read();  k=read();  block=(ll)sqrt(n*1.0);
	up(i,1,n)  a[i]=read();
	up(i,1,m)  q[i].x=read(),q[i].y=read(),q[i].id=i;
}
int main(){
	freopen(FILE".in","r",stdin);
	freopen(FILE".out","w",stdout);
	init();
	Mo_Team::work();
	return 0;
}