【bzoj3781】小B的询问 莫队算法

原文地址：http://www.cnblogs.com/GXZlegend/p/6803821.html

题目描述

小B有一个序列，包含N个1~K之间的整数。他一共有M个询问，每个询问给定一个区间[L..R]，求Sigma(c(i)^2)的值,其中i的值从1到K，其中c(i)表示数字i在[L..R]中的重复次数。小B请你帮助他回答询问。
输入

第一行，三个整数N、M、K。
第二行，N个整数，表示小B的序列。
接下来的M行，每行两个整数L、R。
输出

M行，每行一个整数，其中第i行的整数表示第i个询问的答案。
样例输入

6 4 3

1 3 2 1 1 3

1 4

2 6

3 5

5 6

样例输出

6

9

5

2

题解

莫队算法模板题，优雅的暴力

设原来有n个某颜色，加1后对答案的贡献为(n+1)^2-n^2=2*n+1，减1对答案的贡献为(n-1)^2-n^2=-(2*n-1)。

然后各种区间平移得到答案。

#include <cstdio>
#include <cmath>
#include <algorithm>
#define N 50010
using namespace std;
struct data
{
int l , r , b , p;
}a
;
int c
;
long long ans
, cnt
;
bool cmp(data x , data y)
{
return x.b == y.b ? x.r < y.r : x.b < y.b;
}
int main()
{
int n , m , k , i , si , lp = 1 , rp = 0 , now = 0;
scanf("%d%d%d" , &n , &m , &k);
si = (int)sqrt(n);
for(i = 1 ; i <= n ; i ++ ) scanf("%d" , &c[i]);
for(i = 1 ; i <= m ; i ++ ) scanf("%d%d" , &a[i].l , &a[i].r) , a[i].b = (a[i].l - 1) / si , a[i].p = i;
sort(a + 1 , a + m + 1 , cmp);
for(i = 1 ; i <= m ; i ++ )
{
while(lp < a[i].l) now -= 2 * cnt[c[lp]] - 1 , cnt[c[lp]] -- , lp ++ ;
while(rp > a[i].r) now -= 2 * cnt[c[rp]] - 1 , cnt[c[rp]] -- , rp -- ;
while(lp > a[i].l) lp -- , now += 2 * cnt[c[lp]] + 1 , cnt[c[lp]] ++ ;
while(rp < a[i].r) rp ++ , now += 2 * cnt[c[rp]] + 1 , cnt[c[rp]] ++ ;
ans[a[i].p] = now;
}
for(i = 1 ; i <= m ; i ++ ) printf("%lld\n" , ans[i]);
return 0;
}