【题目描述】
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
1 3 2 1 1 3
1 4
2 6
3 5
5 6
Sample Output
6
9
5
2
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 ",fans[i]); return 0; }