这道题由于数据较大,不能简单地使用一般的树状数组单点更新和单点向下求和,而应该使用区间范围的更新和求和。
不过区间求和的公式证明看得有点久,但最终还是弄懂了,以下为链接:点击打开链接
知道这个以后直接用就好了,不过要注意更新和查询时的左右边界的取值。
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=400000+50;
long long n,m,a[maxn],c1[maxn],c2[maxn],sum[maxn];
char str[3];
long long Lowbit(long long x)
{
return x&(-x);
}
void Add(long long a[],long long x,long long d)
{
while(x<=n)
{
a[x]+=d;
x+=Lowbit(x);
}
}
long long Sum(long long a[],long long x)
{
long long sum=0;
while(x>0)
{
sum+=a[x];
x-=Lowbit(x);
}
return sum;
}
int main()
{
while(scanf("%lld%lld",&n,&m)!=EOF)
{
memset(c1,0,sizeof(c1));
memset(c2,0,sizeof(c2));
memset(sum,0,sizeof(sum));
for(int i=1;i<=n;i++)
scanf("%lld",&a[i]);
sum[0]=0;
for(int j=1;j<=n;j++)
sum[j]=sum[j-1]+a[j];
for(int i=1;i<=m;i++)
{
scanf("%s",str);
if(str[0]=='Q')
{
long long left,right,insuml,insumr,insum;
scanf("%lld%lld",&left,&right);
insuml=sum[left-1]+(left)*Sum(c1,left-1)-Sum(c2,left-1);
insumr=sum[right]+(right+1)*Sum(c1,right)-Sum(c2,right);
insum=insumr-insuml;
printf("%lld\n",insum);
}
else
{
long long left,right,addVal;
scanf("%lld%lld%lld",&left,&right,&addVal);
Add(c1,left,addVal);
Add(c1,right+1,-addVal);
Add(c2,left,left*addVal);
Add(c2,right+1,-(right+1)*addVal);
}
}
}
return 0;
}