bzoj 3781: 小B的询问（莫队）

3781: 小B的询问

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 1064  Solved: 713

[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

很简单的O(1)修改

莫队模板题。。

#include<stdio.h>
#include<math.h>
#include<algorithm>
using namespace std;
#define LL long long
typedef struct Res
{
int x;
int y, id;
}Res;
Res s[100005];
int a[100005], bel[100005];
LL ans[100005], sum[100005];
bool comp(Res a, Res b)
{
if(bel[a.x]<bel[b.x] || bel[a.x]==bel[b.x] && a.y<b.y)
return 1;
return 0;
}
int main(void)
{
LL val;
int n, m, k, i, x, y, B, L, R;
scanf("%d%d%d", &n, &m, &k);
for(i=1;i<=n;i++)
scanf("%d", &a[i]);
for(i=1;i<=m;i++)
{
scanf("%d%d", &s[i].x, &s[i].y);
s[i].id = i;
}
B = sqrt(n)+1;
x = 0, y = 1;
for(i=1;i<=n;i++)
{
x++;
bel[i] = y;
if(x>=B)
x = 0, y++;
}
sort(s+1, s+m+1, comp);
L = R = 1;
sum[a[1]] = val = 1;
for(i=1;i<=m;i++)
{
while(L<s[i].x)
{
val -= (sum[a[L]])*(sum[a[L]]);
sum[a[L]]--;
val += (sum[a[L]])*(sum[a[L]]);
L++;
}
while(L>s[i].x)
{
L--;
val -= (sum[a[L]])*(sum[a[L]]);
sum[a[L]]++;
val += (sum[a[L]])*(sum[a[L]]);
}
while(R<s[i].y)
{
R++;
val -= (sum[a[R]])*(sum[a[R]]);
sum[a[R]]++;
val += (sum[a[R]])*(sum[a[R]]);
}
while(R>s[i].y)
{
val -= (sum[a[R]])*(sum[a[R]]);
sum[a[R]]--;
val += (sum[a[R]])*(sum[a[R]]);
R--;
}
ans[s[i].id] = val;
}
for(i=1;i<=m;i++)
printf("%lld\n", ans[i]);
return 0;
}