莫队练习之基础练习：P2709小B的询问

题意

给定一数列，给定多个询问，询问中给定l、r，求l到r中每个数出现次数的平方的和

思路

询问之间不互相影响，考虑离线，考虑莫队，维护每个数出现的次数cnt[i]，根据完全平方展开，可以得到更新后sum=sum+2*cnt[i]+1，离线时间复杂度O（n√n），查询O（1）

代码

#include<bits/stdc++.h>
#define N 50005
#define ll long long
using namespace std;
inline ll read(){
	ll x=0,f=1;char c=getchar();
	while(!isdigit(c)){if(c=='-')f=-1;c=getchar();}
	while(isdigit(c)){x=x*10+c-'0';c=getchar();}
	return x*f;
}
ll n,m,k,maxn,q[N],cnt[N],sum,ans[N];
struct query{
	ll l,r,id;
	bool operator<(const query &a)const{
		if(l/maxn!=a.l/maxn)return l<a.l;
		else return (l/maxn)&1?r<a.r:r>a.r;
	}
}a[N];
void add(int i){sum+=2*cnt[i]+1;cnt[i]++;}
void del(int i){sum-=2*cnt[i]-1;cnt[i]--;}
void solve(){
	sort(a+1,a+m+1);for(int i=1,l=1,r=0;i<=m;i++){
		if(a[i].l==a[i].r){ans[a[i].id]=1;continue;}
		while(l>a[i].l)add(q[--l]);
		while(r<a[i].r)add(q[++r]);
		while(l<a[i].l)del(q[l++]);
		while(r>a[i].r)del(q[r--]);
		ans[a[i].id]=sum;
	}
}
int main(){
	n=read(),m=read(),k=read(),maxn=sqrt(n);
	for(int i=1;i<=n;i++)q[i]=read();
	for(int i=1;i<=m;i++){ll l=read(),r=read();a[i]=(query){l,r,i};}
	solve();for(int i=1;i<=m;i++)printf("%lld\n",ans[i]);
	return 0;
}

出现的问题

一定一定一定好好读题，千万千万千万不要忘了输入k！！！