题意:
给你N个数,然后有M个操作。有两种类型的操作,
(1)“Ca b c”,表示将区间[a,b]里的数加上c。
(2)“Q a b”,表示查询区间[a,b]的数的和。
分析:
线段树维护区间修改区间求和
代码:
#include<cstdio>
#include<cstring>
using namespace std;
const int N=1001000;
int a[N],n,m,x,y,z;
struct tree{long long l,r,sum,lazy;}g[N];
char s[1000];
void pushup(int rt){g[rt].sum=g[2*rt].sum+g[2*rt+1].sum;}
void pushdown(int rt)
{
if (g[rt].lazy)
{
g[2*rt].sum+=g[rt].lazy*(g[2*rt].r-g[2*rt].l+1);
g[2*rt+1].sum+=g[rt].lazy*(g[2*rt+1].r-g[2*rt+1].l+1);
g[2*rt].lazy+=g[rt].lazy;
g[2*rt+1].lazy+=g[rt].lazy;
g[rt].lazy=0;
}
}
void build(int rt,int l,int r)
{
g[rt].l=l;
g[rt].r=r;
g[rt].sum=0;
g[rt].lazy=0;
if (l==r)
g[rt].sum=a[l]; else
{
int mid=(l+r)/2;
build(2*rt,l,mid);
build(2*rt+1,mid+1,r);
pushup(rt);
}
}
void update(int rt,int ll,int rr,int val)
{
int l=g[rt].l,r=g[rt].r;
if (ll<=l && r<=rr)
{
g[rt].lazy+=val;
g[rt].sum+=val*(r-l+1);
} else
{
pushdown(rt);
int mid=(l+r)/2;
if (ll<=mid) update(2*rt,ll,rr,val);
if (rr> mid) update(2*rt+1,ll,rr,val);
pushup(rt);
}
}
long long query(int rt,int ll,int rr)
{
pushdown(rt);
int l=g[rt].l,r=g[rt].r;
if (ll<=l && r<=rr) return g[rt].sum;
long long mid=(l+r)/2,suml=0,sumr=0;
if (ll<=mid) suml=query(2*rt,ll,rr);
if (rr> mid) sumr=query(2*rt+1,ll,rr);
return suml+sumr;
}
int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++)
scanf("%d",a+i);
build(1,1,n);
for (int i=1;i<=m;i++)
{
scanf("%s",s);
if (s[0]=='Q')
{
scanf("%d%d",&x,&y);
printf("%I64d\n",query(1,x,y));
} else
{
scanf("%d%d%d",&x,&y,&z);
update(1,x,y,z);
}
}
return 0;
}