P3372 【模板】线段树 1 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
#define int long long
int n,m;
const int N=100010;
struct node{
int l,r,sum,lazy;
};
int a[N];
node tr[N*4];
void pushup(int u){
tr[u].sum=tr[u*2].sum+tr[u*2+1].sum;
}
void build(int u,int l,int r){
tr[u]={l,r};
if(l==r){
tr[u].sum=a[l];
tr[u].lazy=0;
return;
}
int mid=(l+r)/2;
build(u*2,l,mid);
build(u*2+1,mid+1,r);
pushup(u);//更新父亲
}
void down(int u,int x){
tr[u].lazy+=x;
tr[u].sum+=x*(tr[u].r-tr[u].l+1);
}
void pushdown(int u){
if(tr[u].lazy==0) return;
down(u*2,tr[u].lazy);
down(u*2+1,tr[u].lazy);
tr[u].lazy=0;
}
/*
修改操作,搜索到根节点修改,然后需要回溯修改所有父亲
*/
void change(int u,int l,int r,int x){
if(l<=tr[u].l&&tr[u].r<=r){//找到根节点
tr[u].sum+=x*(tr[u].r-tr[u].l+1);
tr[u].lazy+=x;
return;
}
pushdown(u);//更新儿子
int mid=(tr[u].l+tr[u].r)/2;
if(l<=mid) change(u*2,l,r,x);
if(r>mid) change(u*2+1,l,r,x);
pushup(u); //更新父亲
}
/*
查询操作,所有儿子都是父亲过来的,不需要回溯父亲
*/
int query(int u,int l,int r){
if(l<=tr[u].l&&tr[u].r<=r) return tr[u].sum;
int mid=(tr[u].l+tr[u].r)/2;
int res=0;
pushdown(u);
if(l<=mid) res+=query(u*2,l,r);
if(r>mid) res+=query(u*2+1,l,r);
return res;
}
void solve()
{
cin>>n>>m;
for(int i=1;i<=n;i++){
cin>>a[i];
}
build(1,1,n);
while(m--){
int s; cin>>s;
if(s==1){
int l,r,k;
cin>>l>>r>>k;
change(1,l,r,k);
}else{
int l,r;
cin>>l>>r;
cout<<query(1,l,r)<<endl;
}
}
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0),cout.tie(0);
solve();
return 0;
}