P2709 小B的询问-CSDN博客

本文链接：https://blog.csdn.net/summ1ts/article/details/141613533

非常简单的莫队板子题，让我们求出区间[l,r]中每个数出现次数的平方和，设枚举到 $a_i$ ,原来答案是res，如果加上 $a_i$ 后，则原来的 $cnt_{a_i}^{2}$ 变为 $(cnt_{a_i}+1)^{2}$ ，即res相比原来加上 $2*cnt_{a_i}+1$ ，删除同理。知道如何维护一个数的添加和删除后，剩下的套莫队板子就行了。

#include<bits/stdc++.h>
using namespace std;
const int N=5e4+10;

int n,m,k,t,a[N],ans[N],pos[N],cnt[N],res;

struct node{
	int l,r,id;
}que[N];

bool cmp(node a,node b){
	if(pos[a.l]==pos[b.l]) return a.r<b.r;
	return a.l<b.l;
}

void del(int x){
	res+=-2*cnt[a[x]]+1;
	cnt[a[x]]--;
}

void add(int x){
	res+=2*cnt[a[x]]+1;
	cnt[a[x]]++;
}

int main(){
	
	cin>>n>>m>>k;for(int i=1;i<=n;i++) cin>>a[i];
	t=sqrt(n);for(int i=1;i<=n;i++) pos[i]=(i-1)/t+1;
	for(int i=1;i<=m;i++) cin>>que[i].l>>que[i].r,que[i].id=i;
	sort(que+1,que+1+m,cmp);
	for(int i=1,l=1,r=0;i<=m;i++){
		while(l<que[i].l) del(l++);
		while(r>que[i].r) del(r--);
		while(l>que[i].l) add(--l);
		while(r<que[i].r) add(++r);
		ans[que[i].id]=res;
	}
	for(int i=1;i<=m;i++) cout<<ans[i]<<endl;
	
	return 0;
}