和上一题差不多。
给出n个数,存在两种操作,Q(打印
∑bi=aAi
),C(
Ai+=c,a=<i<=b
)。
同样使用线段树进行处理
因为long long的问题交了几发不必要的WA..
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
#define MAXN 111111
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define mid int m = (l + r) >> 1
long long sum[MAXN<<2];
long long col[MAXN<<2];
void PushUp(int rt)
{
sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}
void PushDown(int rt,int m)
{
if(col[rt])
{
col[rt<<1] += col[rt];
col[rt<<1|1] += col[rt];
sum[rt<<1] += col[rt] *(m-(m>>1));
sum[rt<<1|1] += col[rt] *(m>>1);
col[rt] = 0;
}
}
void build(int l,int r,int rt)
{
col[rt] = 0;
if(l==r)
{
cin>>sum[rt];
return;
}
mid;
build(lson);
build(rson);
PushUp(rt);
}
void update(int L,int R,long long c,int l,int r,int rt)
{
if(L <= l&& r <= R){
col[rt] += c;
sum[rt] += c * (r - l + 1);
return;
}
PushDown(rt,r-l+1);
mid;
if(L<=m) update(L,R,c,lson);
if(R>m) update(L,R,c,rson);
PushUp(rt);
}
long long query(int L,int R,int l,int r,int rt)
{
long long ret = 0;
mid;
if(L<=l&&r<=R)
return sum[rt];
PushDown(rt,r-l+1);
if(L<=m) ret += query(L,R,lson);
if(R>m) ret += query(L,R,rson);
return ret;
}
int main()
{
ios::sync_with_stdio(false);
int n,q,a,b;
long long c;
char ch;
cin>>n>>q;
build(1,n,1);
while(q--)
{
cin>>ch;
if(ch=='Q')
{
cin>>a>>b;
cout<<query(a,b,1,n,1)<<endl;
}
else
{
cin>>a>>b>>c;
update(a,b,c,1,n,1);
}
}
}