[bzoj3781]小B的询问【莫队】

【题目描述】

Description

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

Input

第一行，三个整数N、M、K。

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

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

Output

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

Sample Input

6 4 3
1 3 2 1 1 3
1 4
2 6
3 5
5 6

Sample Output

6
9
5
2

HINT

对于全部的数据，1<=N、M、K<=50000

Source

【题解】

 莫队裸题，先把数列分成sqrt(n)个块，按对于询问[l..r]，把所有询问按左端点所在块为第一关键字，右端点为第二关键字排序，然后暴力。

 块内的更改，l变化不会超过sqrt(n)，r单调不减，所以一个块内所有r的更改不会超过n，一共sqrt(n)个块。

总复杂度O(n*sqrt(n))

/* --------------
    user Vanisher
    problem bzoj-3781
----------------*/
# include <bits/stdc++.h>
# define 	ll 		long long
# define 	ui 		unsigned int
# define 	N 		50010
using namespace std;
ui read(){
	ui tmp=0, fh=1; char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-') fh=-1; ch=getchar();}
	while (ch>='0'&&ch<='9'){tmp=tmp*10+ch-'0'; ch=getchar();}
	return tmp*fh;
}
struct Q{
	ui id,nl,l,r;
}q[N];
bool cmp(Q x, Q y){
	return x.nl<y.nl||x.nl==y.nl&&x.r<y.r;
}
ui n,m,k,T,lasl,lasr,h[N],ans,sum[N],fans[N];
int main(){
	n=read(); m=read(); k=read(); T=sqrt(n);
	for (ui i=1; i<=n; i++) h[i]=read();
	for (ui i=1; i<=m; i++){
		q[i].l=read(), q[i].r=read(); 
		q[i].nl=q[i].l/T; q[i].id=i;
	}
	sort(q+1,q+m+1,cmp);
	lasl=1; lasr=0;
	for (ui i=1; i<=m; i++){
		if (q[i].r>lasr)
			for (ui j=lasr+1; j<=q[i].r; j++)
				ans=ans+(sum[h[j]]++)*2+1;
			else for (ui j=lasr; j>=q[i].r+1; j--)
				ans=ans-(--sum[h[j]])*2-1;
		lasr=q[i].r;
		if (q[i].l>lasl)
			for (ui j=lasl; j<=q[i].l-1; j++)
				ans=ans-(--sum[h[j]])*2-1;
			else for (ui j=lasl-1; j>=q[i].l; j--)
				ans=ans+(sum[h[j]]++)*2+1;
		lasl=q[i].l;
		fans[q[i].id]=ans;
	}
	for (ui i=1; i<=m; i++)
		printf("%d\n",fans[i]);
	return 0;
}