题意简述
给你
n
个数,
每个函数的意义是数
[L,R]
的和。
q
次操作,每次询问一段函数的和,或更改一个数。
数据范围
1≤ai≤109
思路
对函数分块。
每块建立一个系数数组,表示有多少个
ai
。
每块维护一个答案。
这样我们就可以
O(n√)
完成整块的维护和查询了。
那么散点怎么办?
我们对数在分块。
维护一个前缀和,每块维护一个加标记。
每次修改块内重构,块外打标记。
复杂度
O(qn√)
代码
#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
#define block 310
#define MAXN 100010
typedef unsigned long long ll;
int n,u,v,q,opt,bu,bv;
int l[MAXN],r[MAXN],xi[400][MAXN],a[MAXN];
ll seq[MAXN],add[400],f[400],ans,w;
ll calc(int x)
{
ll ret=seq[r[x]]+add[r[x]/block];
if (l[x])
ret-=seq[l[x]-1]+add[(l[x]-1)/block];
return ret;
}
int main()
{
scanf("%d",&n);
for (int i=0;i<n;i++)
scanf("%llu",&seq[i]);
for (int i=0;i<n;i++)
{
scanf("%d%d",&l[i],&r[i]);
l[i]--,r[i]--;
}
for (int i=0;i*block<n;i++)
{
memset(a,0,sizeof(a));
for (int j=i*block;j<min((i+1)*block,n);j++)
a[l[j]]++,a[r[j]+1]--;
for (int j=1;j<n;j++)
a[j]+=a[j-1];
for (int j=0;j<n;j++)
{
xi[i][j]=a[j];
f[i]+=xi[i][j]*seq[j];
}
}
for (int i=1;i<n;i++)
seq[i]+=seq[i-1];
scanf("%d",&q);
for (int i=0;i<q;i++)
{
scanf("%d",&opt);
if (opt==1)
{
scanf("%d%llu",&u,&w);
u--;
bu=u/block;
w-=((seq[u]+add[bu])-(u!=0)*(seq[u-1]+add[(u-1)/block]));
for (int j=bu*block;j<u;j++)
seq[j]+=add[bu];
for (int j=u;j<min((bu+1)*block,n);j++)
seq[j]+=w+add[bu];
add[bu]=0;
for (int j=bu+1;j*block<n;j++)
add[j]+=w;
for (int j=0;j*block<n;j++)
f[j]+=xi[j][u]*w;
}
if (opt==2)
{
scanf("%d%d",&u,&v);
u--,v--;
bu=u/block,bv=v/block;
ans=0;
if (bu==bv)
for (int j=u;j<=v;j++)
ans+=calc(j);
else
{
for (int j=u;j<min((bu+1)*block,n);j++)
ans+=calc(j);
for (int j=bv*block;j<=v;j++)
ans+=calc(j);
for (int j=bu+1;j<bv;j++)
ans+=f[j];
}
printf("%llu\n",ans);
}
}
return 0;
}