洛谷 P2709 小B的询问 莫队

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接： https://blog.csdn.net/qq_44343213/article/details/102715674

题目描述
小B有一个序列，包含N个1~K之间的整数。他一共有M个询问，每个询问给定一个区间[L…R]，求Sigma(c(i)^2)的值,其中i的值从1到K，其中c(i)表示数字i在[L…R]中的重复次数。小B请你帮助他回答询问。

输入格式
第一行，三个整数N、M、K。

第二行，N个整数，表示小B的序列。

接下来的M行，每行两个整数L、R。

输出格式
M行，每行一个整数，其中第i行的整数表示第i个询问的答案。

输入输出样例
输入 #1
6 4 3
1 3 2 1 1 3
1 4
2 6
3 5
5 6
输出 #1
6
9
5
2
说明/提示
对于全部的数据，1<=N、M、K<=50000

一道不平常的莫队模板题，每次我们移动区间时，需要进行一下数学分析，然后就很简单了

#include<cstdio>
#include<algorithm>
#include<cmath>
#define re register int
#define si 50005
using namespace std;
struct node {
	int l,r,id,pos;
	bool operator < (const node &x) const {
		return pos==x.pos?r<x.r:pos<x.pos;
	}
}e[si];
int n,m,k,block,b[si],cnt[si];
long long ans,Ans[si];
inline int read() {
	int x=0,cf=1;
	char ch=getchar();
	while(ch<'0'||ch>'9') {
		if(ch=='-') cf=-1;
		ch=getchar();
	}
	while(ch>='0'&&ch<='9') {
		x=(x<<3)+(x<<1)+ch-'0';
		ch=getchar();
	}
	return x*cf;
}
int main() {
	n=read(),m=read(),k=read();
	block=(int)sqrt(n);
	for(re i=1;i<=n;i++) b[i]=read();
	for(re i=1;i<=m;i++) {
		e[i].l=read(),e[i].r=read();
		e[i].id=i,e[i].pos=(e[i].l-1)/block+1;
	}
	sort(e+1,e+m+1);
	int l=1,r=0;
	for(re i=1;i<=m;i++) {
		while(l>e[i].l) l--,cnt[b[l]]++,ans+=2*cnt[b[l]]-1;
		while(r<e[i].r) r++,cnt[b[r]]++,ans+=2*cnt[b[r]]-1;
		while(l<e[i].l) cnt[b[l]]--,ans-=2*cnt[b[l]]+1,l++;
		while(r>e[i].r) cnt[b[r]]--,ans-=2*cnt[b[r]]+1,r--;
		Ans[e[i].id]=ans;
	}
	for(re i=1;i<=m;i++) {
		printf("%d\n",Ans[i]);
	}
	return 0;
}