代码如下:
#include<bits/stdc++.h>
#define ls rt<<1
#define rs rt<1|1
#define lson ls,l,mid
#define rson rs,mid+1,r
using namespace std;
long long n,m;
vector<long long> tr,a,lazy;
void push_up(long long rt){
tr[rt]=tr[ls]+tr[rs];
}
void push_down(long long rt,long long l,long long r){
if(lazy[rt]==0) return;
long long mid=l+r>>1;
lazy[ls]+=lazy[rt];
lazy[rs]+=lazy[rt];
tr[ls]+=lazy[rt]*(mid-l+1);
tr[rs]+=lazy[rt]*(r-mid);
lazy[rt]=0;
}
void build(long long rt,long long l,long long r){
lazy[rt]=0;
if(l==r){
tr[rt]=a[l];
return;
}
long long mid=l+r>>1;
build(lson);
build(rson);
push_up(rt);
}
long long query(long long rt,long long l,long long r,long long ql,long long qr){
if(ql<=l&&r<=qr){
return tr[rt];
}
long long mid=l+r>>1;
push_down(rt,l,r);
if(qr<=mid) return query(lson,ql,qr);
if(mid<ql) return query(rson,ql,qr);
return query(lson,ql,qr)+query(rson,ql,qr);
}
void update(long long rt,long long l,long long r,long long ql,long long qr,long long k){
if(ql<=l&&r<=qr){
lazy[rt]+=k;
tr[rt]+=k*(r-l+1);
return;
}
long long mid=l+r>>1;
push_down(rt,l,r);
if(ql<=mid) update(lson,ql,qr,k);
if(mid<qr) update(rson,ql,qr,k);
push_up(rt);
}
int main(){
ios::sync_with_stdio(0);
cin.tie(0);
cin>>n>>m;
lazy=tr=vector<long long>(n*4+1,0);
a=vector<long long>(n+1,0);
for(long long i=1;i<=n;i++)
cin>>a[i];
build(1,1,n);
for(long long i=1;i<=m;i++){
long long op;
cin>>op;
if(op==1){
long long l,r,x;
cin>>l>>r>>x;
update(1,1,n,l,r,x);
}
if(op==2){
long long l,r;
cin>>l>>r;
cout<<query(1,1,n,l,r)<<'\n';
}
}
}