#include<algorithm>
#include<iostream>
#include<cstdio>
#include<string.h>
using namespace std;
const long long maxn=100010;
struct node
{
long long lazy,sum;
long long r,l;
};
node a[maxn<<2];
long long number[maxn];
long long update(long long k)
{
a[k].sum=a[k<<1].sum+a[k<<1|1].sum;
}
void build(long long k,long long l,long long r)
{
a[k].r=r;
a[k].l=l;
a[k].lazy=0;
if(r==l)
{
a[k].sum=number[l];
return ;
}
else
{
long long mid=(l+r)>>1;
build(k<<1,l,mid);
build(k<<1|1,mid+1,r);
update(k);
}
}void down(long long k)
{
if(a[k].l==a[k].r)
{
a[k].lazy=0;
return ;
}
a[k<<1].lazy+=a[k].lazy;
a[k<<1|1].lazy+=a[k].lazy;
a[k<<1].sum+=(a[k<<1].r-a[k<<1].l+1)*a[k].lazy;
a[k<<1|1].sum+=(a[k<<1|1].r-a[k<<1|1].l+1)*a[k].lazy;
a[k].lazy=0;
}
void change(long long k,long long l,long long r,long long y)
{
long long mid;
if(a[k].l==l&&a[k].r==r)
{
a[k].lazy+=y;
a[k].sum+=(r-l+1)*y;
return ;
}
//[1,5]有标记时,修改了 [1,3 ] 的值,如果没有down,在update时 [1,5]的值就错了。
down(k);
mid=(a[k].l+a[k].r)>>1;
if(mid<l)
change(k<<1|1,l,r,y);
else if(mid>=r)
change(k<<1,l,r,y);
else
{
change(k<<1,l,mid,y);
change(k<<1|1,mid+1,r,y);
}
update(k);
}
long long ans=0;
long long query(long long k,long long l,long long r)
{
if(a[k].l==l&&a[k].r==r)
{
return a[k].sum;
}
long long mid=(a[k].r+a[k].l)>>1;
if(a[k].lazy!=0)
{
down(k);
}
long long res = 0;
if(mid<l)
return query(k<<1|1,l,r);
else if(mid>=r)
return query(k<<1,l,r);
else
{
return query(k<<1,l,mid)+query(k<<1|1,mid+1,r);
}
}
int main()
{
long long n,m;
scanf("%lld%lld",&n,&m);
for(long long i=1; i<=n; i++)
{
scanf("%lld",&number[i]);
}
build(1,1,n);
while(m--)
{
char h;
cin>>h;
long long q,w;
long long e;
if(h=='Q')
{
scanf("%lld%lld",&q,&w);
ans=query(1,q,w);
cout<<ans<<endl;
}
else
{
scanf("%lld%lld%lld",&q,&w,&e);
change(1,q,w,e);
}
}
}
POJ 3468 (线段树)
最新推荐文章于 2020-08-08 16:58:26 发布