[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
;
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
,ans,sum
,fans
;
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;
}