题目描述
给定一个长度为N的颜色序列C，对于该序列中的任意一个元素Ci，都有1<=Ci<=M。对于一种颜色ColorK来说，区间[L,R]内的权值定义为这种颜色在该区间中出现的次数的平方，即区间[L,R]内中满足Ci=ColorK的元素个数的平方。接下来给出Q个询问，询问区间[L,R]内颜色[a,b]的权值总和。
输入
第1行三个整数N,M,Q。分别代表序列长度，颜色总数和询问总数。
第2行N个整数，代表序列Ci。
第3行到第Q+2行，每行4个整数l,r,a,b。记上一次计算出的答案为Lans。那么实际的l,r,a,b为给出的l,r,a,b xor上Lans。第一个询问的时候Lans=0。
输出
总共Q行，对于每一个询问，输出权值总和
样例输入
4 2 3
1 1 2 2
1 4 1 2
10 11 9 10
3 0 0 0
样例输出
8
2
0
题解
分块
这种二叉数据结构维护不了，又强制在线的，大概就是分块了。
维护 $f[i][j][k]$ 表示从第 $i$ 块到第 $j$ 块，权值在 $[1,k]$ 之间的个数平方和。为了方便处理零碎部分，还要维护 $c[i][j][k]$ 表示从第 $i$ 块到第 $j$ 块，权值为 $k$ 的个数。
对于询问转化为权值的前缀相减处理，整块部分直接拿出答案，零碎部分暴力枚举，算出对平方和的贡献。
设块的大小为 $si$ ，则预处理时间复杂度为 $O(n·(frac n{si})^2)$ ，询问时间复杂度为 $O(n·si)$ 。根据均值不等式，当 $si=n^{frac 23}$ 时复杂度最优，为 $O(n^{frac 53})$
#include <cstdio>
#include <cstring>
typedef unsigned int ui;
ui a[50010] , f[40][40][20010] , c[40][40][20010] , sum[20010] , cnt[20010];
int main()
{
ui n , m , q , si = 1 , i , j , k , l , r , x , y , bl , br , ans = 0;
scanf("%u%u%u" , &n , &m , &q);
for(i = 1 ; i <= n ; i ++ ) scanf("%d" , &a[i]);
while(si * si * si < n * n) si ++ ;
for(i = 1 ; i <= (n - 1) / si + 1 ; i ++ )
{
for(j = i ; j <= (n - 1) / si + 1 ; j ++ )
{
for(k = (j - 1) * si + 1 ; k <= j * si && k <= n ; k ++ ) sum[a[k]] += cnt[a[k]] << 1 | 1 , cnt[a[k]] ++ ;
for(k = 1 ; k <= m ; k ++ ) c[i][j][k] = cnt[k] , f[i][j][k] = sum[k] + f[i][j][k - 1];
}
memset(sum , 0 , sizeof(sum));
memset(cnt , 0 , sizeof(cnt));
}
while(q -- )
{
scanf("%u%u%u%u" , &l , &r , &x , &y) , l ^= ans , r ^= ans , x ^= ans , y ^= ans;
bl = (l - 1) / si + 1 , br = (r - 1) / si + 1 , ans = f[bl + 1][br - 1][y] - f[bl + 1][br - 1][x - 1];
if(bl == br)
{
for(i = l ; i <= r ; i ++ ) if(a[i] >= x && a[i] <= y) ans += cnt[a[i]] << 1 | 1 , cnt[a[i]] ++ ;
for(i = l ; i <= r ; i ++ ) if(a[i] >= x && a[i] <= y) cnt[a[i]] -- ;
}
else
{
for(i = l ; i <= bl * si ; i ++ ) if(a[i] >= x && a[i] <= y) ans += (c[bl + 1][br - 1][a[i]] + cnt[a[i]]) << 1 | 1 , cnt[a[i]] ++ ;
for(i = r ; i > (br - 1) * si ; i -- ) if(a[i] >= x && a[i] <= y) ans += (c[bl + 1][br - 1][a[i]] + cnt[a[i]]) << 1 | 1 , cnt[a[i]] ++ ;
for(i = l ; i <= bl * si ; i ++ ) if(a[i] >= x && a[i] <= y) cnt[a[i]] -- ;
for(i = r ; i > (br - 1) * si ; i -- ) if(a[i] >= x && a[i] <= y) cnt[a[i]] -- ;
}
printf("%u
" , ans);
}
return 0;
}