题目描述
如题,已知一个数列,你需要进行下面两种操作:
将某区间每一个数加上 kk。
求出某区间每一个数的和。
输入格式
第一行包含两个整数 n, mn,m,分别表示该数列数字的个数和操作的总个数。
第二行包含 nn 个用空格分隔的整数,其中第 ii 个数字表示数列第 ii 项的初始值。
接下来 mm 行每行包含 33 或 44 个整数,表示一个操作,具体如下:
1 x y k:将区间 [x, y] 内每个数加上 kk。
2 x y:输出区间 [x, y] 内每个数的和。
输出格式
输出包含若干行整数,即为所有操作 2 的结果。
输入输出样例
输入
5 5
1 5 4 2 3
2 2 4
1 2 3 2
2 3 4
1 1 5 1
2 1 4
输出
11
8
20
#include<iostream>
#include<cstring>
#include<algorithm>
#include<string>
#include<cstdio>
using namespace std;
typedef long long ll;
const int maxn=100005;
ll a[maxn<<2],tag[maxn<<2],sum[maxn<<2];
ll n,m;
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
void pushup(ll rt){
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void move(ll rt,ll l, ll r,ll t){
sum[rt]+=(r-l+1)*t;
tag[rt]+=t;
}
void pushdown(ll rt,ll l,ll r){
if(!tag[rt])return ;
ll mid=(l+r)>>1;
move(lson,tag[rt]);
move(rson,tag[rt]);
tag[rt]=0;
}
void build(ll rt,ll l,ll r){
if(l==r){
sum[rt]=a[l];
return ;
}
int mid=(l+r)>>1;
build(lson);
build(rson);
pushup(rt);
}
void update(ll ql,ll qr,ll rt,ll l,ll r,ll num){
if(ql<=l&&r<=qr){
sum[rt]+=num*(r-l+1);
tag[rt]+=num;
return ;
}
pushdown(rt,l,r);
ll mid=(l+r)>>1;
if(ql<=mid){
update(ql,qr,lson,num);
}
if(mid<qr){
update(ql,qr,rson,num);
}
pushup(rt);
}
ll query(ll ql,ll qr,ll rt,ll l,ll r){
if(ql<=l&&r<=qr){
return sum[rt];
}
ll mid=(l+r)>>1;
ll res=0;
pushdown(rt,l,r);
if(ql<=mid){
res+=query(ql,qr,lson);
}
if(mid<qr){
res+=query(ql,qr,rson);
}
return res;
}
int main(){
scanf("%lld%lld",&n,&m);
for(int i=1;i<=n;i++){
scanf("%lld",&a[i]);
}
build(1,1,n);
while(m--){
int p;
scanf("%d",&p);
if(p==1){
ll x,y,z;
scanf("%lld%lld%lld",&x,&y,&z);
update(x,y,1,1,n,z);
}
else {
ll x,y;
scanf("%lld%lld",&x,&y);
printf("%lld\n",query(x,y,1,1,n));
}
}
return 0;
}