BZOJ-3781: 小B的询问(莫队算法)

3781: 小B的询问

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 1039  Solved: 697
[Submit][Status][Discuss]

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

哈哈哈也许是上一题搞的时间太长了，老天眷顾了我一道水题，虽然是道权限题，虽然一开始由于轻敌把题目看错了 _(:зゝ∠)_ 然而还是很快就水出来咯~

#include "bits/stdc++.h"
using namespace std;
typedef long long LL;
const int MAX=5e4+5;
int n,m;
LL a[MAX],b[MAX],pos[MAX],ans,an[MAX],bas;
struct Node{
    int id;
    int l,r;
    bool operator < (const Node &tt) const {
        if (pos[l]!=pos[tt.l])
            return pos[l]<pos[tt.l];
        return r<tt.r;
    }
}que[MAX];
inline int read(){
    int an=0,x=1;char c=getchar();
    while (c<'0' || c>'9') {if (c=='-') x=-1;c=getchar();}
    while (c>='0' && c<='9') {an=an*10+c-'0';c=getchar();}
    return an*x;
}
void update(int x,int y){
    ans-=b[a[x]]*b[a[x]];
    b[a[x]]+=y;
    ans+=b[a[x]]*b[a[x]];
}
int main(){
    freopen ("ask.in","r",stdin);
    freopen ("ask.out","w",stdout);
    int i,j;
    n=read();m=read();read();bas=(LL)sqrt(n*1.0);
    memset(b,0,sizeof(b));
    for (i=1;i<=n;i++) a[i]=read(),pos[i]=(i-1)/bas+1;
    for (i=1;i<=m;i++){
        que[i].id=i;
        que[i].l=read();que[i].r=read();
    }
    sort(que+1,que+m+1);
    int L=1,R=0;
    for (i=1;i<=m;i++){
        while (R<que[i].r){R++;update(R,1);}
        while (L>que[i].l){L--;update(L,1);}
        while (R>que[i].r){update(R,-1);R--;}
        while (L<que[i].l){update(L,-1);L++;}
        an[que[i].id]=ans;
    }
    for (i=1;i<=m;i++)
        printf("%d\n",an[i]);
    return 0;
}