bzoj 2906: 颜色 分块

题意

给定一个长度为N的颜色序列C，对于该序列中的任意一个元素Ci，都有1<=Ci<=M。对于一种颜色ColorK来说，区间[L,R]内的权值定义为这种颜色在该区间中出现的次数的平方，即区间[L,R]内中满足Ci=ColorK的元素个数的平方。接下来给出Q个询问，询问区间[L,R]内颜色[a,b]的权值总和。
1<=N,Q<=50000,M<=20000，强制在线

分析

把平方转化成有多少个有序数对(x,y)满足Cx=Cy。
如果不强制在线的话显然可以用莫队来搞，但如果要强制在线的话，我们考虑分块。
预处理ans[i,j,k]表示第i块到第j块中颜色[1,k]的贡献，sum[i,j]表示前i块中有多少个位置颜色为j。然后在询问的时候就可以O(1)得出两个端点之间的块的答案，然后两边剩余的块暴力扫。
设块的大小为B，那么预处理的复杂度是$O(\frac{n^2m}{B^2})$，询问的复杂度是$O(qB)$，解一下方程不难得出$B=(nm)^{\frac{1}{3}}=1000$，那么就做完了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>

typedef long long LL;

const int N=50005;
const int M=52;
const int B=1000;

int n,m,q,sum[M][20005],c[N],bel[N],sta[M],end[M],t[N];
LL ans[M][M][20005];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void pre()
{
    for (int i=1;i<=n;i++)
    {
        bel[i]=(i+B-1)/B;
        end[bel[i]]=i;
        if (!sta[bel[i]]) sta[bel[i]]=i;
    }
    for (int i=1;i<=bel[n];i++)
    {
        for (int j=1;j<=m;j++) sum[i][j]=sum[i-1][j];
        for (int j=sta[i];j<=end[i];j++) sum[i][c[j]]++;
    }
    for (int i=1;i<=bel[n];i++)
    {
        for (int j=i;j<=bel[n];j++)
        {
            for (int k=1;k<=m;k++) ans[i][j][k]=ans[i][j-1][k];
            for (int k=sta[j];k<=end[j];k++) ans[i][j][c[k]]+=t[c[k]]*2+1,t[c[k]]++;
        }
        for (int j=sta[i];j<=n;j++) t[c[j]]=0;
    }
    for (int i=1;i<=bel[n];i++)
        for (int j=i;j<=bel[n];j++)
            for (int k=1;k<=m;k++)
                ans[i][j][k]+=ans[i][j][k-1];
}

LL query(int l,int r,int a,int b)
{
    LL w=ans[bel[l]+1][bel[r]-1][b]-ans[bel[l]+1][bel[r]-1][a-1];
    if (bel[l]==bel[r])
    {
        for (int i=l;i<=r;i++)
            if (c[i]>=a&&c[i]<=b) w+=t[c[i]]*2+1,t[c[i]]++;
        for (int i=l;i<=r;i++) t[c[i]]=0;
        return w;
    }
    for (int i=l;i<=end[bel[l]];i++)
    {
        if (c[i]<a||c[i]>b) continue;
        if (!t[c[i]]) t[c[i]]=sum[bel[r]-1][c[i]]-sum[bel[l]][c[i]];
        w+=t[c[i]]*2+1;t[c[i]]++;
    }
    for (int i=sta[bel[r]];i<=r;i++)
    {
        if (c[i]<a||c[i]>b) continue;
        if (!t[c[i]]) t[c[i]]=sum[bel[r]-1][c[i]]-sum[bel[l]][c[i]];
        w+=t[c[i]]*2+1;t[c[i]]++;
    }
    for (int i=l;i<=end[bel[l]];i++) t[c[i]]=0;
    for (int i=sta[bel[r]];i<=r;i++) t[c[i]]=0;
    return w;
}

int main()
{
    n=read();m=read();q=read();
    for (int i=1;i<=n;i++) c[i]=read();
    pre();
    LL ls=0;
    while (q--)
    {
        int l=read()^ls,r=read()^ls,a=read()^ls,b=read()^ls;
        printf("%lld\n",ls=query(l,r,a,b));
    }
    return 0;
}