BZOJ3781: 小B的询问

3781: 小B的询问

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 146  Solved: 98
[Submit][Status]

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

题解：

裸莫队。。

代码：

 #include<cstdio>

 #include<cstdlib>

 #include<cmath>

 #include<cstring>

 #include<algorithm>

 #include<iostream>

 #include<vector>

 #include<map>

 #include<set>

 #include<queue>

 #include<string>

 #define inf 1000000000

 #define maxn 100000+5

 #define maxm 20000000+5

 #define eps 1e-10

 #define ll long long

 #define pa pair<int,int>

 #define for0(i,n) for(int i=0;i<=(n);i++)

 #define for1(i,n) for(int i=1;i<=(n);i++)

 #define for2(i,x,y) for(int i=(x);i<=(y);i++)

 #define for3(i,x,y) for(int i=(x);i>=(y);i--)

 #define mod 1000000007
 #define sqr(x) (x)*(x)

 using namespace std;

 inline int read()

 {

     int x=,f=;char ch=getchar();

     while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}

     while(ch>=''&&ch<=''){x=*x+ch-'';ch=getchar();}

     return x*f;

 }
 int n,m,k,sum,s[maxn],a[maxn],b[maxn],ans[maxn];
 struct rec{int id,l,r;}c[maxn];
 inline void update(int x,int y)
 {
     sum-=sqr(s[x]);
     s[x]+=y;
     sum+=sqr(s[x]);
 }
 inline bool cmp(const rec &x,const rec &y){return b[x.l]==b[y.l]?x.r<y.r:x.l<y.l;}

 int main()

 {

     freopen("input.txt","r",stdin);

     freopen("output.txt","w",stdout);
     n=read();m=read();k=read();int len=sqrt(n);
     for1(i,n)a[i]=read(),b[i]=(i-)/len+;
     for1(i,m)c[i].l=read(),c[i].r=read(),c[i].id=i;
     sort(c+,c+m+,cmp);
     int l=,r=;
     for1(i,m)
     {
         while(r<c[i].r)update(a[++r],);
         while(r>c[i].r)update(a[r--],-);
         while(l>c[i].l)update(a[--l],);
         while(l<c[i].l)update(a[l++],-);
         ans[c[i].id]=sum;

     }
     for1(i,m)printf("%d\n",ans[i]);

     return ;

 }