题意:给定一个数列,1操作求一个这样的区间[ L , R ]和:a[ R ]+a[ R-1 ]*2+a[ R-2 ]*3+...+a[ L ]*( R - L +1 ),2操作修改一个a[ i ]的值。
思路:借助两颗线段树,线段树sum1维护正常区间和,线段树sum2维护数列:a[ n ],a[ n-1 ]*2,a[ n-2 ]*3,,,,a[ 1 ]*n 的区间和,假如题目求区间[ l ,r ]的的和,先算出 r 到 n 的差 len ,那么sum2[ l , r] - sum1[ l , r ]*len就是答案。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=1e5+10;
typedef long long ll;
ll sum1[maxn*4],sum2[maxn*4],a[maxn];
int n;
void build(int o,int l,int r)
{
if(l==r)
{
sum1[o]=a[l];
sum2[o]=a[l]*(n-l+1);
return;
}
int ls=o*2,rs=o*2+1,m=(l+r)/2;
build(ls,l,m);
build(rs,m+1,r);
sum1[o]=sum1[ls]+sum1[rs];
sum2[o]=sum2[ls]+sum2[rs];
}
ll query(int o,int l,int r,int ql,int qr)
{
if(l>=ql&&r<=qr)return sum2[o];
int ls=o*2,rs=o*2+1,m=(l+r)/2;
ll res=0;
if(ql<=m) res+=query(ls,l,m,ql,qr);
if(qr>m) res+=query(rs,m+1,r,ql,qr);
return res;
}
ll query2(int o,int l,int r,int ql,int qr)
{
if(l>=ql&&r<=qr)return sum1[o];
int ls=o*2,rs=o*2+1,m=(l+r)/2;
ll res=0;
if(ql<=m) res+=query2(ls,l,m,ql,qr);
if(qr>m) res+=query2(rs,m+1,r,ql,qr);
return res;
}
void update(int o,int l,int r,int k,ll v)
{
if(l==r)
{
sum1[o]=v;
sum2[o]=v*(n-l+1);
return;
}
int ls=o*2,rs=o*2+1,m=(l+r)/2;
if(k<=m)update(ls,l,m,k,v) ;
else update(rs,m+1,r,k,v);
sum1[o]=sum1[ls]+sum1[rs];
sum2[o]=sum2[ls]+sum2[rs];
}
int main()
{
int m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)scanf("%lld",&a[i]);
build(1,1,n);
while(m--)
{
int op,l,r;
scanf("%d%d%d",&op,&l,&r);
if(op==1)
{
ll t1=query(1,1,n,l,r);
ll t2=query2(1,1,n,l,r);
int tmp=n-r;
printf("%lld\n",t1-t2*tmp);
}
else
update(1,1,n,l,r);
}
}