题意
给定一个长度为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(n2mB2)
O
(
n
2
m
B
2
)
,询问的复杂度是
O(qB)
O
(
q
B
)
,解一下方程不难得出
B=(nm)13=1000
B
=
(
n
m
)
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;
}